150 lines
4.9 KiB
Python
150 lines
4.9 KiB
Python
from pydantic import BaseModel, EmailStr, Field
|
|
from typing import Optional
|
|
from datetime import datetime
|
|
|
|
|
|
class UserBase(BaseModel):
|
|
"""用户基础模型"""
|
|
|
|
username: str = Field(..., min_length=3, max_length=50, description="登录名")
|
|
nickname: str = Field(..., max_length=50, description="昵称")
|
|
email: Optional[EmailStr] = Field(None, max_length=50, description="电子邮箱")
|
|
phone: Optional[str] = Field(None, max_length=11, description="手机号码")
|
|
wx_openid: Optional[str] = Field(None, max_length=100, description="微信OpenID")
|
|
avatar: Optional[str] = Field(None, max_length=255, description="头像路径")
|
|
role_id: int = Field(2, description="角色ID")
|
|
|
|
|
|
class UserCreateRequest(UserBase):
|
|
"""用户创建请求模型"""
|
|
|
|
password: str = Field(..., min_length=6, max_length=50, description="密码")
|
|
|
|
|
|
class UserCreate(UserBase):
|
|
"""用户创建模型"""
|
|
|
|
password_hash: str = Field(
|
|
..., min_length=6, max_length=128, description="密码哈希值"
|
|
)
|
|
|
|
|
|
class UserUpdate(BaseModel):
|
|
"""用户更新模型"""
|
|
|
|
username: str = Field(..., min_length=3, max_length=50, description="登录名")
|
|
nickname: str = Field(..., max_length=50, description="昵称")
|
|
email: Optional[EmailStr] = Field(None, max_length=50, description="电子邮箱")
|
|
phone: Optional[str] = Field(None, max_length=11, description="手机号码")
|
|
wx_openid: Optional[str] = Field(None, max_length=100, description="微信OpenID")
|
|
avatar: Optional[str] = Field(None, max_length=255, description="头像路径")
|
|
role_id: Optional[int] = Field(2, description="角色ID")
|
|
password: Optional[str] = Field(None, max_length=128, description="密码")
|
|
|
|
|
|
class UserLogin(BaseModel):
|
|
"""用户登录模型"""
|
|
|
|
username: str = Field(..., description="登录名")
|
|
password: str = Field(..., description="密码")
|
|
captcha_id: str = Field(..., description="验证码ID")
|
|
captcha_code: str = Field(..., description="验证码")
|
|
|
|
|
|
class UserResponse(BaseModel):
|
|
"""用户响应模型"""
|
|
|
|
id: int = Field(..., description="用户ID")
|
|
username: str = Field(..., description="登录名")
|
|
nickname: str = Field(..., description="昵称")
|
|
email: Optional[EmailStr] = Field(None, description="电子邮箱")
|
|
phone: Optional[str] = Field(None, description="手机号码")
|
|
wx_openid: Optional[str] = Field(None, description="微信OpenID")
|
|
avatar: Optional[str] = Field(None, description="头像路径")
|
|
role_id: int = Field(..., description="角色ID")
|
|
createtime: datetime = Field(..., description="创建时间")
|
|
updatetime: datetime = Field(..., description="更新时间")
|
|
lastlogintime: Optional[datetime] = Field(None, description="最后登录时间")
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class Token(BaseModel):
|
|
"""令牌模型"""
|
|
|
|
access_token: str = Field(..., description="访问令牌")
|
|
token_type: str = Field(..., description="令牌类型")
|
|
|
|
|
|
class TokenData(BaseModel):
|
|
"""令牌数据模型"""
|
|
|
|
user_id: Optional[int] = None
|
|
email: Optional[EmailStr] = None
|
|
|
|
|
|
class RoleBase(BaseModel):
|
|
"""角色基础模型"""
|
|
|
|
name: str = Field(..., max_length=50, description="角色名称")
|
|
|
|
|
|
class RoleCreate(RoleBase):
|
|
"""角色创建模型"""
|
|
|
|
creator: Optional[str] = Field(None, max_length=50, description="创建人")
|
|
|
|
|
|
class RoleResponse(RoleBase):
|
|
"""角色响应模型"""
|
|
|
|
id: int = Field(..., description="角色ID")
|
|
createtime: datetime = Field(..., description="创建时间")
|
|
updatetime: datetime = Field(..., description="更新时间")
|
|
creator: Optional[str] = Field(None, description="创建人")
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# 权限相关模型
|
|
class PermissionBase(BaseModel):
|
|
"""权限基础模型"""
|
|
|
|
name: str = Field(..., max_length=50, description="权限名称")
|
|
code: str = Field(..., max_length=50, description="权限编码")
|
|
description: Optional[str] = Field(None, max_length=255, description="权限描述")
|
|
|
|
|
|
class PermissionCreate(PermissionBase):
|
|
"""权限创建模型"""
|
|
pass
|
|
|
|
|
|
class PermissionResponse(PermissionBase):
|
|
"""权限响应模型"""
|
|
|
|
id: int = Field(..., description="权限ID")
|
|
createtime: datetime = Field(..., description="创建时间")
|
|
updatetime: datetime = Field(..., description="更新时间")
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# 角色权限关联相关模型
|
|
class RolePermissionAssign(BaseModel):
|
|
"""角色权限分配模型"""
|
|
|
|
role_id: int = Field(..., description="角色ID")
|
|
permission_ids: list[int] = Field(..., description="权限ID列表")
|
|
|
|
|
|
class RolePermissionResponse(BaseModel):
|
|
"""角色权限响应模型"""
|
|
|
|
role_id: int = Field(..., description="角色ID")
|
|
role_name: str = Field(..., description="角色名称")
|
|
permissions: list[PermissionResponse] = Field(..., description="权限列表")
|