HimoolERP/apps/system/models.py

from extensions.exceptions import ValidationError
from extensions.models import *


class Team(Model):

    number = CharField(max_length=32, unique=True, verbose_name='编号')
    expiry_time = DateTimeField(verbose_name='到期时间')
    create_time = DateTimeField(auto_now_add=True, verbose_name='创建时间')


class PermissionType(Model):
    """权限类型"""

    name = CharField(max_length=64, verbose_name='类型名称')


class Permission(Model):
    """权限"""

    type = ForeignKey('system.PermissionType', on_delete=CASCADE, related_name='permissions', verbose_name='权限类型')
    name = CharField(max_length=64, verbose_name='权限名称')
    code = CharField(max_length=64, verbose_name='权限代码')


class Role(Model, ModelMixin):
    """角色"""

    name = CharField(max_length=64, verbose_name='名称')
    remark = CharField(max_length=256, blank=True, null=True, verbose_name='备注')
    permissions = ManyToManyField('system.Permission', blank=True, related_name='roles', verbose_name='权限')
    team = ForeignKey('system.Team', on_delete=CASCADE, related_name='roles')

    class Meta:
        unique_together = [('name', 'team')]


class User(Model, ModelMixin):
    """用户"""

    username = CharField(max_length=32, verbose_name='用户名')
    password = CharField(max_length=256, verbose_name='密码')
    name = CharField(max_length=64, verbose_name='名称')
    phone = CharField(max_length=32, blank=True, null=True, verbose_name='手机号')
    email = CharField(max_length=256, blank=True, null=True, verbose_name='邮箱')
    roles = ManyToManyField('system.Role', blank=True, related_name='users', verbose_name='角色')
    is_manager = BooleanField(default=False, verbose_name='管理员状态')
    is_active = BooleanField(default=True, verbose_name='激活状态')
    create_time = DateTimeField(auto_now_add=True, verbose_name='创建时间')
    team = ForeignKey('system.Team', on_delete=CASCADE, related_name='users')

    class Meta:
        unique_together = [('username', 'team')]

    def validate(self):
        if not self.is_active:
            raise ValidationError(f'用户[{self.username}]未激活')


__all__ = [
    'Team', 'PermissionType', 'Permission', 'Role', 'User',
]