ecoloop/orgs/models.py

from __future__ import annotations

from django.contrib.auth import get_user_model
from django.contrib.sites.models import Site
from django.db import models


User = get_user_model()


class TimestampedModel(models.Model):
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    class Meta:
        abstract = True


class Organization(TimestampedModel):
    name = models.CharField(max_length=255)
    code = models.CharField(max_length=32, unique=True)

    # Minimal curated choices for simplicity; expand as needed
    TZ_CHOICES = (
        ("UTC", "UTC"),
        ("Asia/Bangkok", "Asia/Bangkok"),
        ("Asia/Singapore", "Asia/Singapore"),
        ("Asia/Jakarta", "Asia/Jakarta"),
        ("Europe/Berlin", "Europe/Berlin"),
        ("America/New_York", "America/New_York"),
    )
    timezone = models.CharField(max_length=64, choices=TZ_CHOICES, default="UTC")

    CURRENCY_CHOICES = (
        ("USD", "USD — US Dollar"),
        ("THB", "THB — Thai Baht"),
        ("EUR", "EUR — Euro"),
        ("SGD", "SGD — Singapore Dollar"),
        ("IDR", "IDR — Indonesian Rupiah"),
    )
    currency_code = models.CharField(max_length=8, choices=CURRENCY_CHOICES, default="USD")

    def __str__(self) -> str:
        return self.name


class OrganizationSite(models.Model):
    organization = models.ForeignKey(Organization, on_delete=models.CASCADE, related_name="sites")
    site = models.OneToOneField(Site, on_delete=models.CASCADE, related_name="organization_site")

    def __str__(self) -> str:
        return f"{self.site.domain} -> {self.organization.code}"


class UserProfile(TimestampedModel):
    ROLE_OWNER = "owner"
    ROLE_MANAGER = "manager"
    ROLE_DRIVER = "driver"
    ROLE_CUSTOMER = "customer"
    ROLE_AUDITOR = "auditor"
    ROLE_CHOICES = (
        (ROLE_OWNER, "Owner"),
        (ROLE_MANAGER, "Manager"),
        (ROLE_DRIVER, "Driver"),
        (ROLE_CUSTOMER, "Customer"),
        (ROLE_AUDITOR, "Auditor"),
    )

    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name="recycle_profile")
    organization = models.ForeignKey(Organization, on_delete=models.CASCADE, related_name="users")
    role = models.CharField(max_length=16, choices=ROLE_CHOICES, default=ROLE_MANAGER)
    my_photo = models.ImageField(upload_to="user_photos/", blank=True, null=True)

    def __str__(self) -> str:
        return f"{self.user.username} ({self.role})"