You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.5 KiB
97 lines
3.5 KiB
6 months ago
|
from django.db import models
|
||
|
|
from django.contrib.auth.models import AbstractBaseUser,BaseUserManager, PermissionsMixin
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
from django.conf import settings
|
||
|
|
from uuid import uuid4
|
||
|
|
from django.urls import reverse
|
||
|
|
from .utils import rename_profile_image
|
||
|
|
|
||
|
|
User = settings.AUTH_USER_MODEL
|
||
|
|
|
||
|
|
|
||
|
|
class CustomUserManager(BaseUserManager):
|
||
|
|
def create_user(self, email, first_name, last_name, password=None, **kwargs):
|
||
|
|
if not email:
|
||
|
|
raise ValueError(_('Users must have an email address.'))
|
||
|
|
|
||
|
|
user = self.model(
|
||
|
|
email=self.normalize_email(email),
|
||
|
|
first_name=first_name,
|
||
|
|
last_name=last_name,
|
||
|
|
**kwargs
|
||
|
|
)
|
||
|
|
user.set_password(password)
|
||
|
|
user.save(using=self._db)
|
||
|
|
return user
|
||
|
|
|
||
|
|
def create_superuser(self, email, first_name, last_name, password=None):
|
||
|
|
user = self.create_user(
|
||
|
|
email,
|
||
|
|
password = password,
|
||
|
|
first_name = first_name,
|
||
|
|
last_name = last_name,
|
||
|
|
)
|
||
|
|
user.is_admin = True
|
||
|
|
user.is_verified = True
|
||
|
|
user.is_staff =True
|
||
|
|
user.is_active = True
|
||
|
|
user.is_superuser = True
|
||
|
|
user.save(using=self._db)
|
||
|
|
return user
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class CustomUser(AbstractBaseUser, PermissionsMixin):
|
||
|
|
user = models.OneToOneField(User, null=True, blank= True, on_delete=models.CASCADE)
|
||
|
|
username = models.CharField(max_length=50, unique=True, null = True)
|
||
|
|
first_name = models.CharField(_('first name'),max_length=150, null=True)
|
||
|
|
last_name = models.CharField(_('last name'),max_length=150, null=True)
|
||
|
|
|
||
|
|
email = models.CharField(_('email address'), max_length=150, unique=True)
|
||
|
|
is_email_confirmed = models.BooleanField(default=False)
|
||
|
|
email_confirmation_token = models.CharField(max_length=100, blank=True, null=True)
|
||
|
|
email_confirmation_token_expiry = models.DateTimeField(null=True, blank=True)
|
||
|
|
|
||
|
|
|
||
|
|
phone_number = models.CharField(max_length=15, null = True, blank=True)
|
||
|
|
is_phone_number_confirmed = models.BooleanField(default=False)
|
||
|
|
phone_number_confirmation_token = models.CharField(max_length=50, blank=True, null=True)
|
||
|
|
profile_image = models.ImageField(default='user.png', upload_to=rename_profile_image)
|
||
|
|
|
||
|
|
|
||
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
||
|
|
updated_at = models.DateTimeField(auto_now=True)
|
||
|
|
|
||
|
|
is_verified = models.BooleanField(default=False)
|
||
|
|
is_admin = models.BooleanField(default=False)
|
||
|
|
is_active = models.BooleanField(default=True)
|
||
|
|
is_staff = models.BooleanField(default=False)
|
||
|
|
is_superuser = models.BooleanField(default=False)
|
||
|
|
|
||
|
|
objects = CustomUserManager() # setting customer user namanger to the account
|
||
|
|
|
||
|
|
USERNAME_FIELD = 'email'
|
||
|
|
REQUIRED_FIELDS = [ 'first_name','last_name',]
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
if not self.email:
|
||
|
|
return ""
|
||
|
|
return self.email
|
||
|
|
|
||
|
|
def save(self, *args, **kwargs):
|
||
|
|
if not self.username:
|
||
|
|
# Auto-generate a random username with "user-" prefix
|
||
|
|
self.username = f"user-{str(uuid4())[:8]}"
|
||
|
|
super().save(*args, **kwargs)
|
||
|
|
|
||
|
|
def get_username_display(self):
|
||
|
|
if self.username.startswith('user-'):
|
||
|
|
return self.username[len('user-'):]
|
||
|
|
return self.username
|
||
|
|
|
||
|
|
def get_user_full_name(self):
|
||
|
|
full_name = f"{self.first_name} {self.last_name}"
|
||
|
|
return full_name
|
||
|
|
|
||
|
|
def get_absolute_url_username(self):
|
||
|
|
return reverse('lots:seller_detail', kwargs={'username': self.username})
|