75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
from pydantic import BaseModel, EmailStr, Field
|
|
from typing import Optional
|
|
|
|
|
|
class UserBase(BaseModel):
|
|
"""用户基础数据结构，供多处复用。"""
|
|
|
|
username: str = Field(..., description="用户名")
|
|
email: Optional[EmailStr] = Field(default=None, description="邮箱，可选")
|
|
|
|
|
|
class UserCreate(UserBase):
|
|
"""注册时使用的模型。"""
|
|
|
|
password: str = Field(..., min_length=6, description="明文密码")
|
|
|
|
|
|
class UserUpdate(BaseModel):
|
|
"""用户信息修改模型。"""
|
|
|
|
email: Optional[EmailStr] = Field(default=None, description="邮箱")
|
|
password: Optional[str] = Field(default=None, min_length=6, description="新密码")
|
|
|
|
|
|
class User(UserBase):
|
|
"""对外暴露的用户信息。"""
|
|
|
|
id: int = Field(..., description="用户主键")
|
|
is_admin: bool = Field(default=False, description="是否为管理员")
|
|
must_change_password: bool = Field(default=False, description="是否需要强制修改密码")
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class UserInDB(User):
|
|
"""数据库内部使用的模型，包含哈希后的密码。"""
|
|
|
|
hashed_password: str
|
|
|
|
|
|
class Token(BaseModel):
|
|
"""登录成功后返回的访问令牌。"""
|
|
|
|
access_token: str
|
|
token_type: str = "bearer"
|
|
must_change_password: bool = Field(default=False, description="是否需要强制修改密码")
|
|
|
|
|
|
class TokenPayload(BaseModel):
|
|
"""JWT 负载信息。"""
|
|
|
|
sub: str
|
|
is_admin: bool = False
|
|
|
|
|
|
class UserRegistration(UserCreate):
|
|
"""注册接口需要的字段，包含邮箱验证码。"""
|
|
|
|
verification_code: str = Field(..., min_length=4, max_length=10, description="邮箱验证码")
|
|
|
|
|
|
class PasswordChangeRequest(BaseModel):
|
|
"""管理员修改密码请求模型。"""
|
|
|
|
old_password: str = Field(..., min_length=6, description="当前密码")
|
|
new_password: str = Field(..., min_length=8, description="新密码")
|
|
|
|
|
|
class AuthOptions(BaseModel):
|
|
"""认证相关开关信息，供前端动态控制功能。"""
|
|
|
|
allow_registration: bool = Field(..., description="是否允许开放用户注册")
|
|
enable_linuxdo_login: bool = Field(..., description="是否启用 Linux.do 登录")
|