first

apps/cuser/models.py

@@ -0,0 +1,82 @@
+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}'