30 lines
906 B
Python
30 lines
906 B
Python
|
from typing import List, Optional
|
||
|
|
|
||
|
|
from pydantic import Field
|
||
|
|
from tortoise.contrib.pydantic import pydantic_model_creator
|
||
|
|
|
||
|
|
from models import UserModel, UserRoleModel
|
||
|
|
from schemas.common import QueryData
|
||
|
|
|
||
|
|
UserRead = pydantic_model_creator(UserModel, name="UserOut", exclude=("password",))
|
||
|
|
UserIn = pydantic_model_creator(UserModel, name="UserIn", exclude_readonly=True)
|
||
|
|
|
||
|
|
UserRole = pydantic_model_creator(UserRoleModel, name="UserRole", exclude_readonly=True)
|
||
|
|
|
||
|
|
|
||
|
|
class UserInfo(UserRead):
|
||
|
|
active_rid: int = Field(..., description="用户当前激活角色")
|
||
|
|
rids: List[int] = Field(..., description="用户拥有角色")
|
||
|
|
|
||
|
|
|
||
|
|
class UserAdd(UserIn):
|
||
|
|
rids: List[int] = Field(..., description="用户角色列表")
|
||
|
|
|
||
|
|
|
||
|
|
class UserQuery(QueryData):
|
||
|
|
username: Optional[str] = Field("", description="用户名")
|
||
|
|
nickname: Optional[str] = Field("", description="姓名")
|
||
|
|
|
||
|
|
|
||
|
|
UserList = List[UserRead]
|