69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from datetime import datetime
|
||
from typing import Optional
|
||
from pydantic import BaseModel, EmailStr, Field, ConfigDict
|
||
from app.domain.models.role import UserRole
|
||
|
||
# ============================================
|
||
# Request Schemas (输入)
|
||
# ============================================
|
||
|
||
class UserCreate(BaseModel):
|
||
"""用户注册"""
|
||
username: str = Field(..., min_length=3, max_length=50,
|
||
description="用户名，3-50个字符")
|
||
email: EmailStr = Field(..., description="邮箱地址")
|
||
password: str = Field(..., min_length=6, max_length=100,
|
||
description="密码，至少6个字符")
|
||
role: UserRole = Field(default=UserRole.USER, description="用户角色")
|
||
|
||
class UserLogin(BaseModel):
|
||
"""用户登录"""
|
||
username: str = Field(..., description="用户名或邮箱")
|
||
password: str = Field(..., description="密码")
|
||
|
||
class UserUpdate(BaseModel):
|
||
"""用户信息更新"""
|
||
email: Optional[EmailStr] = None
|
||
password: Optional[str] = Field(None, min_length=6, max_length=100)
|
||
role: Optional[UserRole] = None
|
||
is_active: Optional[bool] = None
|
||
|
||
# ============================================
|
||
# Response Schemas (输出)
|
||
# ============================================
|
||
|
||
class UserResponse(BaseModel):
|
||
"""用户信息响应（不含密码）"""
|
||
id: int
|
||
username: str
|
||
email: str
|
||
role: UserRole
|
||
is_active: bool
|
||
is_superuser: bool
|
||
created_at: datetime
|
||
updated_at: datetime
|
||
|
||
model_config = ConfigDict(from_attributes=True)
|
||
|
||
class UserInDB(UserResponse):
|
||
"""数据库中的用户（含密码哈希）"""
|
||
hashed_password: str
|
||
|
||
# ============================================
|
||
# Token Schemas
|
||
# ============================================
|
||
|
||
class Token(BaseModel):
|
||
"""JWT Token 响应"""
|
||
access_token: str
|
||
refresh_token: Optional[str] = None
|
||
token_type: str = "bearer"
|
||
expires_in: int = Field(..., description="过期时间（秒）")
|
||
|
||
class TokenPayload(BaseModel):
|
||
"""JWT Token Payload"""
|
||
sub: str = Field(..., description="用户ID或用户名")
|
||
exp: Optional[int] = None
|
||
iat: Optional[int] = None
|
||
type: str = Field(default="access", description="token类型: access 或 refresh")
|