|
|
@ -1,12 +1,13 @@ |
|
|
|
import re |
|
|
|
from pydantic import BaseModel, ConfigDict, model_validator |
|
|
|
from pydantic import BaseModel, Field, ConfigDict, model_validator |
|
|
|
from pydantic.alias_generators import to_camel |
|
|
|
from typing import Union, Optional, List |
|
|
|
from typing import Union, Optional, List, Literal |
|
|
|
from datetime import datetime |
|
|
|
from module_admin.entity.vo.role_vo import RoleModel |
|
|
|
from module_admin.entity.vo.dept_vo import DeptModel |
|
|
|
from module_admin.entity.vo.post_vo import PostModel |
|
|
|
from module_admin.annotation.pydantic_annotation import as_query, as_form |
|
|
|
from module_admin.annotation.validate_annotation import NetWork, NotBlank, Size, Xss |
|
|
|
from exceptions.exception import ModelValidatorException |
|
|
|
|
|
|
|
|
|
|
@ -14,7 +15,7 @@ class TokenData(BaseModel): |
|
|
|
""" |
|
|
|
token解析结果 |
|
|
|
""" |
|
|
|
user_id: Union[int, None] = None |
|
|
|
user_id: Union[int, None] = Field(default=None, description='用户ID') |
|
|
|
|
|
|
|
|
|
|
|
class UserModel(BaseModel): |
|
|
@ -23,26 +24,26 @@ class UserModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True) |
|
|
|
|
|
|
|
user_id: Optional[int] = None |
|
|
|
dept_id: Optional[int] = None |
|
|
|
user_name: Optional[str] = None |
|
|
|
nick_name: Optional[str] = None |
|
|
|
user_type: Optional[str] = None |
|
|
|
email: Optional[str] = None |
|
|
|
phonenumber: Optional[str] = None |
|
|
|
sex: Optional[str] = None |
|
|
|
avatar: Optional[str] = None |
|
|
|
password: Optional[str] = None |
|
|
|
status: Optional[str] = None |
|
|
|
del_flag: Optional[str] = None |
|
|
|
login_ip: Optional[str] = None |
|
|
|
login_date: Optional[datetime] = 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 |
|
|
|
user_id: Optional[int] = Field(default=None, description='用户ID') |
|
|
|
dept_id: Optional[int] = Field(default=None, description='部门ID') |
|
|
|
user_name: Optional[str] = Field(default=None, description='用户账号') |
|
|
|
nick_name: Optional[str] = Field(default=None, description='用户昵称') |
|
|
|
user_type: Optional[str] = Field(default=None, description='用户类型(00系统用户)') |
|
|
|
email: Optional[str] = Field(default=None, description='用户邮箱') |
|
|
|
phonenumber: Optional[str] = Field(default=None, description='手机号码') |
|
|
|
sex: Optional[Literal['0', '1', '2']] = Field(default=None, description='用户性别(0男 1女 2未知)') |
|
|
|
avatar: Optional[str] = Field(default=None, description='头像地址') |
|
|
|
password: Optional[str] = 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代表删除)') |
|
|
|
login_ip: Optional[str] = Field(default=None, description='最后登录IP') |
|
|
|
login_date: Optional[datetime] = Field(default=None, description='最后登录时间') |
|
|
|
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') |
|
|
|
|
|
|
|
@model_validator(mode='after') |
|
|
|
def check_password(self) -> 'UserModel': |
|
|
@ -60,6 +61,32 @@ class UserModel(BaseModel): |
|
|
|
self.admin = False |
|
|
|
return self |
|
|
|
|
|
|
|
@Xss(field_name='user_name', message='用户账号不能包含脚本字符') |
|
|
|
@NotBlank(field_name='user_name', message='用户账号不能为空') |
|
|
|
@Size(field_name='user_name', min_length=0, max_length=30, message='用户账号长度不能超过30个字符') |
|
|
|
def get_user_name(self): |
|
|
|
return self.user_name |
|
|
|
|
|
|
|
@Xss(field_name='nick_name', message='用户昵称不能包含脚本字符') |
|
|
|
@Size(field_name='nick_name', min_length=0, max_length=30, message='用户昵称长度不能超过30个字符') |
|
|
|
def get_nick_name(self): |
|
|
|
return self.nick_name |
|
|
|
|
|
|
|
@NetWork(field_name='email', field_type='EmailStr', message='邮箱格式不正确') |
|
|
|
@Size(field_name='email', min_length=0, max_length=50, message='邮箱长度不能超过50个字符') |
|
|
|
def get_email(self): |
|
|
|
return self.email |
|
|
|
|
|
|
|
@Size(field_name='phonenumber', min_length=0, max_length=11, message='手机号码长度不能超过11个字符') |
|
|
|
def get_phonenumber(self): |
|
|
|
return self.phonenumber |
|
|
|
|
|
|
|
def validate_fields(self): |
|
|
|
self.get_user_name() |
|
|
|
self.get_nick_name() |
|
|
|
self.get_email() |
|
|
|
self.get_phonenumber() |
|
|
|
|
|
|
|
|
|
|
|
class UserRoleModel(BaseModel): |
|
|
|
""" |
|
|
@ -67,8 +94,8 @@ class UserRoleModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True) |
|
|
|
|
|
|
|
user_id: Optional[int] = None |
|
|
|
role_id: Optional[int] = None |
|
|
|
user_id: Optional[int] = Field(default=None, description='用户ID') |
|
|
|
role_id: Optional[int] = Field(default=None, description='角色ID') |
|
|
|
|
|
|
|
|
|
|
|
class UserPostModel(BaseModel): |
|
|
@ -77,23 +104,23 @@ class UserPostModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True) |
|
|
|
|
|
|
|
user_id: Optional[int] = None |
|
|
|
post_id: Optional[int] = None |
|
|
|
user_id: Optional[int] = Field(default=None, description='用户ID') |
|
|
|
post_id: Optional[int] = Field(default=None, description='岗位ID') |
|
|
|
|
|
|
|
|
|
|
|
class UserInfoModel(UserModel): |
|
|
|
post_ids: Optional[Union[str, None]] = None |
|
|
|
role_ids: Optional[Union[str, None]] = None |
|
|
|
dept: Optional[Union[DeptModel, None]] = None |
|
|
|
role: Optional[List[Union[RoleModel, None]]] = [] |
|
|
|
post_ids: Optional[Union[str, None]] = Field(default=None, description='岗位ID信息') |
|
|
|
role_ids: Optional[Union[str, None]] = Field(default=None, description='角色ID信息') |
|
|
|
dept: Optional[Union[DeptModel, None]] = Field(default=None, description='部门信息') |
|
|
|
role: Optional[List[Union[RoleModel, None]]] = Field(default=[], description='角色信息') |
|
|
|
|
|
|
|
|
|
|
|
class CurrentUserModel(BaseModel): |
|
|
|
model_config = ConfigDict(alias_generator=to_camel) |
|
|
|
|
|
|
|
permissions: List |
|
|
|
roles: List |
|
|
|
user: Union[UserInfoModel, None] |
|
|
|
permissions: List = Field(description='权限信息') |
|
|
|
roles: List = Field(description='角色信息') |
|
|
|
user: Union[UserInfoModel, None] = Field(description='用户信息') |
|
|
|
|
|
|
|
|
|
|
|
class UserDetailModel(BaseModel): |
|
|
@ -102,11 +129,11 @@ class UserDetailModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel) |
|
|
|
|
|
|
|
data: Optional[Union[UserInfoModel, None]] = None |
|
|
|
post_ids: Optional[List] = None |
|
|
|
posts: List[Union[PostModel, None]] |
|
|
|
role_ids: Optional[List] = None |
|
|
|
roles: List[Union[RoleModel, None]] |
|
|
|
data: Optional[Union[UserInfoModel, None]] = Field(default=None, description='用户信息') |
|
|
|
post_ids: Optional[List] = Field(default=None, description='岗位ID信息') |
|
|
|
posts: List[Union[PostModel, None]] = Field(description='岗位信息') |
|
|
|
role_ids: Optional[List] = Field(default=None, description='角色ID信息') |
|
|
|
roles: List[Union[RoleModel, None]] = Field(description='角色信息') |
|
|
|
|
|
|
|
|
|
|
|
class UserProfileModel(BaseModel): |
|
|
@ -115,17 +142,17 @@ class UserProfileModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel) |
|
|
|
|
|
|
|
data: Union[UserInfoModel, None] |
|
|
|
post_group: Union[str, None] |
|
|
|
role_group: Union[str, None] |
|
|
|
data: Union[UserInfoModel, None] = Field(description='用户信息') |
|
|
|
post_group: Union[str, None] = Field(description='岗位信息') |
|
|
|
role_group: Union[str, None] = Field(description='角色信息') |
|
|
|
|
|
|
|
|
|
|
|
class UserQueryModel(UserModel): |
|
|
|
""" |
|
|
|
用户管理不分页查询模型 |
|
|
|
""" |
|
|
|
begin_time: Optional[str] = None |
|
|
|
end_time: Optional[str] = None |
|
|
|
begin_time: Optional[str] = Field(default=None, description='开始时间') |
|
|
|
end_time: Optional[str] = Field(default=None, description='结束时间') |
|
|
|
|
|
|
|
|
|
|
|
@as_query |
|
|
@ -134,24 +161,24 @@ class UserPageQueryModel(UserQueryModel): |
|
|
|
""" |
|
|
|
用户管理分页查询模型 |
|
|
|
""" |
|
|
|
page_num: int = 1 |
|
|
|
page_size: int = 10 |
|
|
|
page_num: int = Field(default=1, description='当前页码') |
|
|
|
page_size: int = Field(default=10, description='每页记录数') |
|
|
|
|
|
|
|
|
|
|
|
class AddUserModel(UserModel): |
|
|
|
""" |
|
|
|
新增用户模型 |
|
|
|
""" |
|
|
|
role_ids: Optional[List] = [] |
|
|
|
post_ids: Optional[List] = [] |
|
|
|
type: Optional[str] = None |
|
|
|
role_ids: Optional[List] = Field(default=[], description='角色ID信息') |
|
|
|
post_ids: Optional[List] = Field(default=[], description='岗位ID信息') |
|
|
|
type: Optional[str] = Field(default=None, description='操作类型') |
|
|
|
|
|
|
|
|
|
|
|
class EditUserModel(AddUserModel): |
|
|
|
""" |
|
|
|
编辑用户模型 |
|
|
|
""" |
|
|
|
role: Optional[List] = [] |
|
|
|
role: Optional[List] = Field(default=[], description='角色信息') |
|
|
|
|
|
|
|
|
|
|
|
@as_query |
|
|
@ -161,8 +188,8 @@ class ResetPasswordModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel) |
|
|
|
|
|
|
|
old_password: Optional[str] = None |
|
|
|
new_password: Optional[str] = None |
|
|
|
old_password: Optional[str] = Field(default=None, description='旧密码') |
|
|
|
new_password: Optional[str] = Field(default=None, description='新密码') |
|
|
|
|
|
|
|
@model_validator(mode='after') |
|
|
|
def check_new_password(self) -> 'ResetPasswordModel': |
|
|
@ -177,9 +204,9 @@ class ResetUserModel(UserModel): |
|
|
|
""" |
|
|
|
重置用户密码模型 |
|
|
|
""" |
|
|
|
old_password: Optional[str] = None |
|
|
|
sms_code: Optional[str] = None |
|
|
|
session_id: Optional[str] = None |
|
|
|
old_password: Optional[str] = Field(default=None, description='旧密码') |
|
|
|
sms_code: Optional[str] = Field(default=None, description='验证码') |
|
|
|
session_id: Optional[str] = Field(default=None, description='会话id') |
|
|
|
|
|
|
|
|
|
|
|
class DeleteUserModel(BaseModel): |
|
|
@ -188,16 +215,16 @@ class DeleteUserModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel) |
|
|
|
|
|
|
|
user_ids: str |
|
|
|
update_by: Optional[str] = None |
|
|
|
update_time: Optional[datetime] = None |
|
|
|
user_ids: str = Field(description='需要删除的用户ID') |
|
|
|
update_by: Optional[str] = Field(default=None, description='更新者') |
|
|
|
update_time: Optional[datetime] = Field(default=None, description='更新时间') |
|
|
|
|
|
|
|
|
|
|
|
class UserRoleQueryModel(UserModel): |
|
|
|
""" |
|
|
|
用户角色关联管理不分页查询模型 |
|
|
|
""" |
|
|
|
role_id: Optional[int] = None |
|
|
|
role_id: Optional[int] = Field(default=None, description='角色ID') |
|
|
|
|
|
|
|
|
|
|
|
@as_query |
|
|
@ -205,15 +232,15 @@ class UserRolePageQueryModel(UserRoleQueryModel): |
|
|
|
""" |
|
|
|
用户角色关联管理分页查询模型 |
|
|
|
""" |
|
|
|
page_num: int = 1 |
|
|
|
page_size: int = 10 |
|
|
|
page_num: int = Field(default=1, description='当前页码') |
|
|
|
page_size: int = Field(default=10, description='每页记录数') |
|
|
|
|
|
|
|
|
|
|
|
class SelectedRoleModel(RoleModel): |
|
|
|
""" |
|
|
|
是否选择角色模型 |
|
|
|
""" |
|
|
|
flag: Optional[bool] = False |
|
|
|
flag: Optional[bool] = Field(default=False, description='选择标识') |
|
|
|
|
|
|
|
|
|
|
|
class UserRoleResponseModel(BaseModel): |
|
|
@ -222,8 +249,8 @@ class UserRoleResponseModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel) |
|
|
|
|
|
|
|
roles: List[Union[SelectedRoleModel, None]] = [] |
|
|
|
user: UserInfoModel |
|
|
|
roles: List[Union[SelectedRoleModel, None]] = Field(default=[], description='角色信息') |
|
|
|
user: UserInfoModel = Field(description='用户信息') |
|
|
|
|
|
|
|
|
|
|
|
@as_query |
|
|
@ -233,7 +260,7 @@ class CrudUserRoleModel(BaseModel): |
|
|
|
""" |
|
|
|
model_config = ConfigDict(alias_generator=to_camel) |
|
|
|
|
|
|
|
user_id: Optional[int] = None |
|
|
|
user_ids: Optional[str] = None |
|
|
|
role_id: Optional[int] = None |
|
|
|
role_ids: Optional[str] = None |
|
|
|
user_id: Optional[int] = Field(default=None, description='用户ID') |
|
|
|
user_ids: Optional[str] = Field(default=None, description='用户ID信息') |
|
|
|
role_id: Optional[int] = Field(default=None, description='角色ID') |
|
|
|
role_ids: Optional[str] = Field(default=None, description='角色ID信息') |
|
|
|