You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
158 lines
5.6 KiB
158 lines
5.6 KiB
from datetime import datetime
|
|
from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
|
|
from pydantic.alias_generators import to_camel
|
|
from pydantic_validation_decorator import NotBlank, Size
|
|
from typing import List, Literal, Optional, Union
|
|
from module_admin.annotation.pydantic_annotation import as_query
|
|
|
|
|
|
class RoleModel(BaseModel):
|
|
"""
|
|
角色表对应pydantic模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
|
|
|
role_id: Optional[int] = Field(default=None, description='角色ID')
|
|
role_name: Optional[str] = Field(default=None, description='角色名称')
|
|
role_key: Optional[str] = Field(default=None, description='角色权限字符串')
|
|
role_sort: Optional[int] = Field(default=None, description='显示顺序')
|
|
data_scope: Optional[Literal['1', '2', '3', '4', '5']] = Field(
|
|
default=None,
|
|
description='数据范围（1：全部数据权限 2：自定数据权限 3：本部门数据权限 4：本部门及以下数据权限 5：仅本人数据权限）',
|
|
)
|
|
menu_check_strictly: Optional[Union[int, bool]] = Field(default=None, description='菜单树选择项是否关联显示')
|
|
dept_check_strictly: Optional[Union[int, bool]] = Field(default=None, description='部门树选择项是否关联显示')
|
|
status: Optional[Literal['0', '1']] = Field(default=None, description='角色状态（0正常 1停用）')
|
|
del_flag: Optional[Literal['0', '2']] = Field(default=None, description='删除标志（0代表存在 2代表删除）')
|
|
create_by: Optional[str] = Field(default=None, description='创建者')
|
|
create_time: Optional[datetime] = Field(default=None, description='创建时间')
|
|
update_by: Optional[str] = Field(default=None, description='更新者')
|
|
update_time: Optional[datetime] = Field(default=None, description='更新时间')
|
|
remark: Optional[str] = Field(default=None, description='备注')
|
|
admin: Optional[bool] = Field(default=False, description='是否为admin')
|
|
|
|
@field_validator('menu_check_strictly', 'dept_check_strictly')
|
|
@classmethod
|
|
def check_filed_mapping(cls, v: Union[int, bool]) -> Union[int, bool]:
|
|
if v == 1:
|
|
v = True
|
|
elif v == 0:
|
|
v = False
|
|
elif v is True:
|
|
v = 1
|
|
elif v is False:
|
|
v = 0
|
|
return v
|
|
|
|
@model_validator(mode='after')
|
|
def check_admin(self) -> 'RoleModel':
|
|
if self.role_id == 1:
|
|
self.admin = True
|
|
else:
|
|
self.admin = False
|
|
return self
|
|
|
|
@NotBlank(field_name='role_name', message='角色名称不能为空')
|
|
@Size(field_name='role_name', min_length=0, max_length=30, message='角色名称长度不能超过30个字符')
|
|
def get_role_name(self):
|
|
return self.role_name
|
|
|
|
@NotBlank(field_name='role_key', message='权限字符不能为空')
|
|
@Size(field_name='role_key', min_length=0, max_length=100, message='权限字符长度不能超过100个字符')
|
|
def get_role_key(self):
|
|
return self.role_key
|
|
|
|
@NotBlank(field_name='role_sort', message='显示顺序不能为空')
|
|
def get_role_sort(self):
|
|
return self.role_sort
|
|
|
|
def validate_fields(self):
|
|
self.get_role_name()
|
|
self.get_role_key()
|
|
self.get_role_sort()
|
|
|
|
|
|
class RoleMenuModel(BaseModel):
|
|
"""
|
|
角色和菜单关联表对应pydantic模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
|
|
|
role_id: Optional[int] = Field(default=None, description='角色ID')
|
|
menu_id: Optional[int] = Field(default=None, description='菜单ID')
|
|
|
|
|
|
class RoleDeptModel(BaseModel):
|
|
"""
|
|
角色和部门关联表对应pydantic模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
|
|
|
role_id: Optional[int] = Field(default=None, description='角色ID')
|
|
dept_id: Optional[int] = Field(default=None, description='部门ID')
|
|
|
|
|
|
class RoleQueryModel(RoleModel):
|
|
"""
|
|
角色管理不分页查询模型
|
|
"""
|
|
|
|
begin_time: Optional[str] = Field(default=None, description='开始时间')
|
|
end_time: Optional[str] = Field(default=None, description='结束时间')
|
|
|
|
|
|
@as_query
|
|
class RolePageQueryModel(RoleQueryModel):
|
|
"""
|
|
角色管理分页查询模型
|
|
"""
|
|
|
|
page_num: int = Field(default=1, description='当前页码')
|
|
page_size: int = Field(default=10, description='每页记录数')
|
|
|
|
|
|
class RoleMenuQueryModel(BaseModel):
|
|
"""
|
|
角色菜单查询模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
menus: List = Field(default=[], description='菜单信息')
|
|
checked_keys: List[int] = Field(default=[], description='已选择的菜单ID信息')
|
|
|
|
|
|
class RoleDeptQueryModel(BaseModel):
|
|
"""
|
|
角色部门查询模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
depts: List = Field(default=[], description='部门信息')
|
|
checked_keys: List[int] = Field(default=[], description='已选择的部门ID信息')
|
|
|
|
|
|
class AddRoleModel(RoleModel):
|
|
"""
|
|
新增角色模型
|
|
"""
|
|
|
|
dept_ids: List = Field(default=[], description='部门ID信息')
|
|
menu_ids: List = Field(default=[], description='菜单ID信息')
|
|
type: Optional[str] = Field(default=None, description='操作类型')
|
|
|
|
|
|
class DeleteRoleModel(BaseModel):
|
|
"""
|
|
删除角色模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
role_ids: str = Field(description='需要删除的菜单ID')
|
|
update_by: Optional[str] = Field(default=None, description='更新者')
|
|
update_time: Optional[datetime] = Field(default=None, description='更新时间')
|
|
|