47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
|
|
from utils.base_model import BaseModel
|
|
import uuid
|
|
|
|
|
|
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):
|
|
username = models.CharField(max_length=50, null=False, blank=False, unique=True)
|
|
mobile = models.PositiveBigIntegerField(null=False, blank=False, unique=True)
|
|
name_family = models.CharField(max_length=100, null=True, blank=True)
|
|
is_active = models.BooleanField(default=False)
|
|
is_staff = models.BooleanField(default=False)
|
|
is_superuser = models.BooleanField(default=False)
|
|
uuid = models.UUIDField(default=uuid.uuid4, editable=False, unique=True)
|
|
|
|
USERNAME_FIELD = 'username'
|
|
|
|
objects = CUserManager()
|
|
|
|
class Meta:
|
|
permissions = [
|
|
('user_is_workspace_owner', 'user is workspace owner'),
|
|
('user_is_workspace_admin', 'user is workspace admin'),
|
|
('user_is_workspace_operator', 'user is workspace operator'),
|
|
('user_is_workspace_customer', 'user is workspace customer'),
|
|
]
|
|
|
|
def __str__(self):
|
|
return f'{self.name_family} - {self.username}'
|