from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin from django.db.models import ( AutoField, BooleanField, CharField, DateField, DateTimeField, EmailField, UUIDField, Model, TextChoices, TextField, URLField, ) from django.utils.translation import gettext_lazy as _ from typing import ClassVar from uuid import uuid4 from apps.users.managers import UserManager from django.conf import settings class TimeStampMixin(Model): created_at = DateTimeField(verbose_name="Created At", auto_now_add=True) updated_at = DateTimeField(verbose_name="Updated At", auto_now=True) class Meta: abstract = True class User(AbstractBaseUser, TimeStampMixin, PermissionsMixin): class Roles(TextChoices): MANAGER = 'manager', _("Manager") EMPLOYEE = 'employee', _("Employee") id = AutoField(verbose_name = _("User ID"), primary_key = True) uuid = UUIDField(verbose_name = _("User UUID"), default = uuid4, editable = False) email_address = EmailField(verbose_name = _("Email Address"), max_length = 255, unique = True) first_name = CharField(verbose_name = _("First Name"), max_length = 255) last_name = CharField(verbose_name = _("Last Name"), max_length = 255) date_of_birth = DateField(verbose_name = _("Date of Birth"), null = True, blank = True) bio = TextField(default = "", blank = True) timezone = CharField(default = settings.TIME_ZONE, max_length = 16, blank = True) avatar_url = URLField(blank = True) is_active = BooleanField(verbose_name = _("Account Active"), default = True) is_staff = BooleanField(verbose_name = _("Account Admin"), default = False) role = CharField(verbose_name = _("Role"), max_length = 50, choices = Roles.choices, default = Roles.EMPLOYEE) USERNAME_FIELD = 'email_address' EMAIL_FIELD = 'email_address' REQUIRED_FIELDS = ['first_name', 'last_name', 'date_of_birth'] objects: ClassVar[UserManager] = UserManager() def has_perm(self, perm, obj=None): return True def has_module_perms(self, app_label): return True class Meta: verbose_name = _('User') verbose_name_plural = _('Users') @property def full_name(self): return f"{self.first_name} {self.last_name}" def __str__(self): return self.full_name