from django.db import models
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
from utils.base_model import BaseModel
import uuid
# Create your models here.


class CUserManager(BaseUserManager):
    def create_user(self, username, password, **extra_fields):
        if not username:
            raise ValueError("Please Enter username")
        user = self.model(username=username, **extra_fields)
        user.set_password(password)
        user.save()
        return user

    def create_superuser(self, username, password, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)
        extra_fields.setdefault('is_active', True)

        return self.create_user(username, password, **extra_fields)


class CUser(BaseModel, AbstractBaseUser, PermissionsMixin):

    STEP_CHOICES = [
        ("1", "1-register"),
        ("2", "2-profileOk")
    ]

    TYPE_CHOICES = [
        ("MO", "mojri"),
        ("OP", "operator")
    ]

    STATUS_CHOICES = [
        ("WT", "waiting"),
        ("AP", "approv"),
        ("RJ", "reject"),
    ]

    username = models.CharField(max_length=50, null=False, blank=False, unique=True)
    mobile = models.PositiveBigIntegerField(null=True, blank=True, unique=True)
    name_family = models.CharField(max_length=50, null=True, blank=True)
    is_active = models.BooleanField(default=False)
    is_staff = models.BooleanField(default=False)
    is_superuser = models.BooleanField(default=False)
    step = models.CharField(default='1', null=False, blank=False, choices=STEP_CHOICES)
    type = models.CharField(default='MO', null=False, blank=False, choices=TYPE_CHOICES)
    mojri_status = models.CharField(default='WT', null=True, blank=True, choices=STATUS_CHOICES)
    uuid = models.UUIDField(default=uuid.uuid4, editable=False, unique=True)

    USERNAME_FIELD = 'username'

    objects = CUserManager()

    def __str__(self):
        return f'{self.name_family} - {self.username} - {self.mobile} - {self.type}'
    
    class Meta:
        ordering = ['-created_at']