You've already forked opc-backend
xujl
23855ef0e4
开发了权限控制接口：实现了通过数据库分配菜单权限节点，控制接口访问安全。 开发了实名认证中心：实现了个人身份证信息与企业营业执照的提交与审核接口。 开发了任务与协作大厅核心业务：实现了任务的发布、接单、状态流转以及专家邀约接口。 配置了全局环境变量与数据库引擎：集成了 PostgreSQL 数据库、Redis 缓存与 MinIO 对象存储。
161 lines
7.0 KiB
Python
161 lines
7.0 KiB
Python
import uuid
|
|
from django.db import models
|
|
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
|
|
|
|
class UserStatus(models.TextChoices):
|
|
ACTIVE = 'ACTIVE', '正常'
|
|
INACTIVE = 'INACTIVE', '未激活'
|
|
BANNED = 'BANNED', '封禁'
|
|
|
|
class CustomUserManager(BaseUserManager):
|
|
def create_user(self, username, phone=None, password=None, **extra_fields):
|
|
if not username:
|
|
raise ValueError('The Username field must be set')
|
|
user = self.model(username=username, phone=phone, **extra_fields)
|
|
if password:
|
|
user.set_password(password)
|
|
user.save(using=self._db)
|
|
return user
|
|
|
|
def create_superuser(self, username, phone=None, password=None, **extra_fields):
|
|
extra_fields.setdefault('is_staff', True)
|
|
extra_fields.setdefault('is_superuser', True)
|
|
return self.create_user(username, phone, password, **extra_fields)
|
|
|
|
class User(AbstractBaseUser, PermissionsMixin):
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
username = models.CharField(max_length=64, unique=True)
|
|
phone = models.CharField(max_length=20, unique=True, null=True, blank=True)
|
|
email = models.EmailField(max_length=128, unique=True, null=True, blank=True)
|
|
nickname = models.CharField(max_length=64, null=True, blank=True)
|
|
avatar_url = models.CharField(max_length=512, null=True, blank=True)
|
|
face_url = models.CharField(max_length=512, null=True, blank=True)
|
|
wx_openid = models.CharField(max_length=128, unique=True, null=True, blank=True)
|
|
wx_unionid = models.CharField(max_length=128, unique=True, null=True, blank=True)
|
|
bio = models.TextField(null=True, blank=True)
|
|
location = models.CharField(max_length=128, null=True, blank=True)
|
|
rating = models.DecimalField(max_digits=3, decimal_places=2, default=5.00)
|
|
completed_tasks = models.IntegerField(default=0)
|
|
|
|
status = models.CharField(max_length=16, choices=UserStatus.choices, default=UserStatus.ACTIVE)
|
|
|
|
is_deleted = models.BooleanField(default=False)
|
|
is_recommended = models.BooleanField(default=False, help_text='管理员推荐')
|
|
recommend_priority = models.IntegerField(default=0, help_text='推荐优先级, 越大越靠前')
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
is_active = models.BooleanField(default=True)
|
|
is_staff = models.BooleanField(default=False)
|
|
|
|
objects = CustomUserManager()
|
|
|
|
@property
|
|
def roles(self):
|
|
return list(self.user_roles.values_list('role__code', flat=True))
|
|
|
|
USERNAME_FIELD = 'username'
|
|
|
|
REQUIRED_FIELDS = ['phone']
|
|
|
|
class Meta:
|
|
db_table = 'users'
|
|
|
|
class Role(models.Model):
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
code = models.CharField(max_length=64, unique=True)
|
|
name = models.CharField(max_length=64)
|
|
description = models.TextField(null=True, blank=True)
|
|
is_system = models.BooleanField(default=False)
|
|
sort_order = models.IntegerField(default=0)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
db_table = 'roles'
|
|
|
|
class PermissionType(models.TextChoices):
|
|
MENU = 'MENU', '菜单'
|
|
BUTTON = 'BUTTON', '按钮'
|
|
API = 'API', '接口'
|
|
|
|
class Permission(models.Model):
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
code = models.CharField(max_length=128, unique=True)
|
|
name = models.CharField(max_length=128)
|
|
type = models.CharField(max_length=16, choices=PermissionType.choices)
|
|
path = models.CharField(max_length=256, null=True, blank=True)
|
|
method = models.CharField(max_length=16, null=True, blank=True)
|
|
parent = models.ForeignKey('self', on_delete=models.SET_NULL, null=True, blank=True)
|
|
sort_order = models.IntegerField(default=0)
|
|
icon = models.CharField(max_length=64, null=True, blank=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
db_table = 'permissions'
|
|
|
|
class UserRole(models.Model):
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='user_roles')
|
|
role = models.ForeignKey(Role, on_delete=models.CASCADE)
|
|
granted_by = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, related_name='granted_roles')
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
db_table = 'user_roles'
|
|
unique_together = ('user', 'role')
|
|
|
|
class RolePermission(models.Model):
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
role = models.ForeignKey(Role, on_delete=models.CASCADE, related_name='role_permissions')
|
|
permission = models.ForeignKey(Permission, on_delete=models.CASCADE)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
db_table = 'role_permissions'
|
|
unique_together = ('role', 'permission')
|
|
|
|
class Enterprise(models.Model):
|
|
class EnterpriseStatus(models.TextChoices):
|
|
PENDING = 'PENDING', '待审核'
|
|
VERIFIED = 'VERIFIED', '已认证'
|
|
REJECTED = 'REJECTED', '已拒绝'
|
|
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
user = models.OneToOneField(User, on_delete=models.CASCADE)
|
|
company_name = models.CharField(max_length=256)
|
|
credit_code = models.CharField(max_length=64, unique=True, null=True, blank=True)
|
|
status = models.CharField(max_length=16, choices=EnterpriseStatus.choices, default=EnterpriseStatus.PENDING)
|
|
|
|
logo_url = models.CharField(max_length=512, null=True, blank=True)
|
|
|
|
business_license = models.CharField(max_length=512, null=True, blank=True)
|
|
contact_name = models.CharField(max_length=64, null=True, blank=True)
|
|
contact_phone = models.CharField(max_length=20, null=True, blank=True)
|
|
landline = models.CharField(max_length=32, null=True, blank=True, help_text='座机号码')
|
|
contact_email = models.CharField(max_length=128, null=True, blank=True)
|
|
address = models.TextField(null=True, blank=True)
|
|
description = models.TextField(null=True, blank=True)
|
|
is_deleted = models.BooleanField(default=False)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
db_table = 'enterprises'
|
|
|
|
class EnterpriseMember(models.Model):
|
|
class MemberRole(models.TextChoices):
|
|
ADMIN = 'ADMIN', '管理员'
|
|
MEMBER = 'MEMBER', '普通成员'
|
|
GUEST = 'GUEST', '外部观察员'
|
|
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
enterprise = models.ForeignKey(Enterprise, on_delete=models.CASCADE, related_name='members')
|
|
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='enterprise_memberships')
|
|
role = models.CharField(max_length=16, choices=MemberRole.choices, default=MemberRole.MEMBER)
|
|
joined_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
db_table = 'enterprise_members'
|
|
unique_together = ('enterprise', 'user')
|
|
|