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.
127 lines
3.1 KiB
127 lines
3.1 KiB
from pydantic import BaseModel, ConfigDict, field_validator, model_validator
|
|
from pydantic.alias_generators import to_camel
|
|
from typing import Union, Optional, List
|
|
from datetime import datetime
|
|
from module_admin.annotation.pydantic_annotation import as_query, as_form
|
|
|
|
|
|
class RoleModel(BaseModel):
|
|
"""
|
|
角色表对应pydantic模型
|
|
"""
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
|
|
|
role_id: Optional[int] = None
|
|
role_name: Optional[str] = None
|
|
role_key: Optional[str] = None
|
|
role_sort: Optional[int] = None
|
|
data_scope: Optional[str] = None
|
|
menu_check_strictly: Optional[Union[int, bool]] = None
|
|
dept_check_strictly: Optional[Union[int, bool]] = None
|
|
status: Optional[str] = None
|
|
del_flag: Optional[str] = None
|
|
create_by: Optional[str] = None
|
|
create_time: Optional[datetime] = None
|
|
update_by: Optional[str] = None
|
|
update_time: Optional[datetime] = None
|
|
remark: Optional[str] = None
|
|
admin: Optional[bool] = False
|
|
|
|
@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
|
|
|
|
|
|
class RoleMenuModel(BaseModel):
|
|
"""
|
|
角色和菜单关联表对应pydantic模型
|
|
"""
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
|
|
|
role_id: Optional[int] = None
|
|
menu_id: Optional[int] = None
|
|
|
|
|
|
class RoleDeptModel(BaseModel):
|
|
"""
|
|
角色和部门关联表对应pydantic模型
|
|
"""
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
|
|
|
role_id: Optional[int] = None
|
|
dept_id: Optional[int] = None
|
|
|
|
|
|
class RoleQueryModel(RoleModel):
|
|
"""
|
|
角色管理不分页查询模型
|
|
"""
|
|
begin_time: Optional[str] = None
|
|
end_time: Optional[str] = None
|
|
|
|
|
|
@as_query
|
|
@as_form
|
|
class RolePageQueryModel(RoleQueryModel):
|
|
"""
|
|
角色管理分页查询模型
|
|
"""
|
|
page_num: int
|
|
page_size: int
|
|
|
|
|
|
class RoleMenuQueryModel(BaseModel):
|
|
"""
|
|
角色菜单查询模型
|
|
"""
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
menus: List = []
|
|
checked_keys: List[int] = []
|
|
|
|
|
|
class RoleDeptQueryModel(BaseModel):
|
|
"""
|
|
角色部门查询模型
|
|
"""
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
depts: List = []
|
|
checked_keys: List[int] = []
|
|
|
|
|
|
class AddRoleModel(RoleModel):
|
|
"""
|
|
新增角色模型
|
|
"""
|
|
dept_ids: List = []
|
|
menu_ids: List = []
|
|
type: Optional[str] = None
|
|
|
|
|
|
class DeleteRoleModel(BaseModel):
|
|
"""
|
|
删除角色模型
|
|
"""
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
role_ids: str
|
|
update_by: Optional[str] = None
|
|
update_time: Optional[datetime] = None
|
|
|