guzhi/app/models/admin.py
2024-07-31 14:58:26 +08:00

93 lines
4.2 KiB
Python

from tortoise import fields
from app.schemas.menus import MenuType
from .base import BaseModel, TimestampMixin
from .enums import MethodType
class User(BaseModel, TimestampMixin):
username = fields.CharField(max_length=20, unique=True, description="用户名称")
alias = fields.CharField(max_length=30, null=True, description="姓名")
email = fields.CharField(max_length=255, unique=True, description="邮箱")
phone = fields.CharField(max_length=20, null=True, description="电话")
password = fields.CharField(max_length=128, null=True, description="密码")
is_active = fields.BooleanField(default=True, description="是否激活")
is_superuser = fields.BooleanField(default=False, description="是否为超级管理员")
last_login = fields.DatetimeField(null=True, description="最后登录时间")
roles = fields.ManyToManyField("models.Role", related_name="user_roles")
dept_id = fields.IntField(null=True, description="部门ID")
class Meta:
table = "user"
class PydanticMeta:
# todo
# computed = ["full_name"]
...
class Role(BaseModel, TimestampMixin):
name = fields.CharField(max_length=20, unique=True, description="角色名称")
desc = fields.CharField(max_length=500, null=True, blank=True, description="角色描述")
menus = fields.ManyToManyField("models.Menu", related_name="role_menus")
apis = fields.ManyToManyField("models.Api", related_name="role_apis")
class Meta:
table = "role"
class Api(BaseModel, TimestampMixin):
path = fields.CharField(max_length=100, description="API路径")
method = fields.CharEnumField(MethodType, description="请求方法")
summary = fields.CharField(max_length=500, description="请求简介")
tags = fields.CharField(max_length=100, description="API标签")
class Meta:
table = "api"
class Menu(BaseModel, TimestampMixin):
name = fields.CharField(max_length=20, description="菜单名称")
remark = fields.JSONField(null=True, description="保留字段", blank=True)
menu_type = fields.CharEnumField(MenuType, null=True, blank=True, description="菜单类型")
icon = fields.CharField(max_length=100, null=True, blank=True, description="菜单图标")
path = fields.CharField(max_length=100, description="菜单路径")
order = fields.IntField(default=0, description="排序")
parent_id = fields.IntField(default=0, max_length=10, description="父菜单ID")
is_hidden = fields.BooleanField(default=False, description="是否隐藏")
component = fields.CharField(max_length=100, description="组件")
keepalive = fields.BooleanField(default=True, description="存活")
redirect = fields.CharField(max_length=100, null=True, blank=True, description="重定向")
class Meta:
table = "menu"
class Dept(BaseModel, TimestampMixin):
name = fields.CharField(max_length=20, unique=True, description="部门名称")
desc = fields.CharField(max_length=500, null=True, blank=True, description="备注")
is_deleted = fields.BooleanField(default=False, description="软删除标记")
order = fields.IntField(default=0, description="排序")
parent_id = fields.IntField(default=0, max_length=10, description="父部门ID")
class Meta:
table = "dept"
class DeptClosure(BaseModel, TimestampMixin):
ancestor = fields.IntField(description="父代")
descendant = fields.IntField(description="子代")
level = fields.IntField(default=0, description="深度")
class AuditLog(BaseModel, TimestampMixin):
user_id = fields.IntField(description="用户ID", index=True)
username = fields.CharField(max_length=64, default="", description="用户名称", index=True)
module = fields.CharField(max_length=64, default="", description="功能模块", index=True)
summary = fields.CharField(max_length=128, default="", description="请求描述", index=True)
method = fields.CharField(max_length=10, default="", description="请求方法", index=True)
path = fields.CharField(max_length=255, default="", description="请求路径", index=True)
status = fields.IntField(default=-1, description="状态码", index=True)
response_time = fields.IntField(default=0, description="响应时间(单位ms)")