83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
|
|
from utils.base_model import BaseModel
|
|
#from apps.workspace.models import Workspace as WorkspaceModel
|
|
import uuid
|
|
|
|
# Create your models here.
|
|
import random
|
|
|
|
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=50, 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 str(self.username)
|
|
|
|
|
|
class RoleType(BaseModel):
|
|
role_name = models.CharField(max_length=20, null=False, blank=False)
|
|
role_name_en = models.CharField(max_length=20, null=False, blank=False, default='')
|
|
|
|
def __str__(self):
|
|
return str(self.role_name)
|
|
|
|
|
|
class CuserRole(BaseModel):
|
|
cuser = models.ForeignKey(CUser, on_delete=models.CASCADE, related_name='cuser_role')
|
|
role_type = models.ForeignKey(RoleType, on_delete=models.CASCADE, related_name='role_type')
|
|
name_family = models.CharField(max_length=30, null=False, blank=False)
|
|
title = models.CharField(max_length=30, null=True, blank=True)
|
|
mobile = models.PositiveBigIntegerField(null=False, blank=False)
|
|
mobile_verify = models.BooleanField(default=False)
|
|
#sys_ws = models.ForeignKey(WorkspaceModel, on_delete=models.CASCADE, null=True, blank=True, default=None)
|
|
|
|
uuid = models.UUIDField(default=uuid.uuid4, editable=False, unique=True)
|
|
is_active = models.BooleanField(default=False)
|
|
|
|
def __str__(self):
|
|
return f'{str(self.cuser) } - { str(self.id) }'
|
|
|
|
|
|
class SmsOpt(BaseModel):
|
|
mobile = models.BigIntegerField(null=False, blank=False)
|
|
opt = models.IntegerField(null=False, blank=False)
|
|
expire = models.DateTimeField(null=True, blank=True)
|
|
|
|
def __str__(self):
|
|
return f'{self.mobile} {self.opt}'
|