63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
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']
|
|
|