from django.db import models
from django.contrib.auth.models import AbstractBaseUser,BaseUserManager,PermissionsMixin
from phonenumber_field.modelfields import PhoneNumberField
from address.models import AddressField
from django.utils.translation import gettext_lazy as _

STATUS =(
    ('Company_Account', _('Company account')),
    ('Need_verification', _("Wait for verification")),
    ('BaseAcc',_('Base account')),
    ('Moderator',_('Moderator')),
)

class UserManager(BaseUserManager):
    def create_user(self,email, password = None, **extra_fields):
        if not email:
            raise ValueError("Users must havve an email address")

        email = self.model (
            email = self.normalize_email(email)
        )
        user = self.model(email = email,**extra_fields)
        user.set_password(password)
        user.save(using = self._db)
        return user

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

        if extra_fields.get('is_staff') is not True:
            raise ValueError(_("Superuser must have is_staff = True"))
        if extra_fields.get('is_superuser') is not True:
            raise ValueError(_("Superuser must have is_superuser = True"))
        return self.create_user(email,password, **extra_fields)

 

class Base_User(AbstractBaseUser,PermissionsMixin):
    id = models.AutoField(primary_key= True)
    email = models.EmailField(_('Email'), unique= True)
    avatar = models.ImageField(upload_to='media/avatars', null = True, blank = True,default='media/avatars/default.png')

    first_name = models.CharField(_('First Name'), max_length=30, blank = False, default = '')
    last_name = models.CharField(_('Last name'), max_length=30, blank = False,default = '')
    avatar = models.ImageField(upload_to='media/avatars', null = True, blank = True,)

    
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)

    date_joined = models.DateTimeField(_('Date joined'),auto_now_add=True)
    status = models.CharField(max_length=32, choices=STATUS, default='BaseAcc')

    USERNAME_FIELD = 'email'

    objects = UserManager()

    def __str__(self):
        return self.email

    class Meta:
        verbose_name = "User"
        verbose_name_plural = "Users"
    
 
class OwnerUser(models.Model):
    user = models.OneToOneField(Base_User, on_delete=models.CASCADE,null = True)
    company_name = models.CharField(_('Company name'), max_length=30, blank= True)