69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
|
"""User schemas."""
|
|||
|
|
|
|||
|
|
from typing import Optional
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
|
|||
|
|
from utils.util_schemas import BaseResponse
|
|||
|
|
|
|||
|
|
class UserBase(BaseModel):
|
|||
|
|
"""User base schema."""
|
|||
|
|
username: str = Field(..., min_length=3, max_length=50)
|
|||
|
|
email: str = Field(..., max_length=100)
|
|||
|
|
full_name: Optional[str] = Field(None, max_length=100)
|
|||
|
|
bio: Optional[str] = None
|
|||
|
|
avatar_url: Optional[str] = None
|
|||
|
|
|
|||
|
|
class UserCreate(UserBase):
|
|||
|
|
"""User creation schema."""
|
|||
|
|
password: str = Field(..., min_length=6)
|
|||
|
|
|
|||
|
|
class UserUpdate(BaseModel):
|
|||
|
|
"""User update schema."""
|
|||
|
|
username: Optional[str] = Field(None, min_length=3, max_length=50)
|
|||
|
|
email: Optional[str] = Field(None, max_length=100)
|
|||
|
|
full_name: Optional[str] = Field(None, max_length=100)
|
|||
|
|
bio: Optional[str] = None
|
|||
|
|
avatar_url: Optional[str] = None
|
|||
|
|
password: Optional[str] = Field(None, min_length=6)
|
|||
|
|
is_active: Optional[bool] = None
|
|||
|
|
|
|||
|
|
class ChangePasswordRequest(BaseModel):
|
|||
|
|
"""Change password request schema."""
|
|||
|
|
current_password: str = Field(..., description="Current password")
|
|||
|
|
new_password: str = Field(..., min_length=6, description="New password")
|
|||
|
|
|
|||
|
|
class ResetPasswordRequest(BaseModel):
|
|||
|
|
"""Admin reset password request schema."""
|
|||
|
|
new_password: str = Field(..., min_length=6, description="New password")
|
|||
|
|
|
|||
|
|
class UserResponse(BaseResponse, UserBase):
|
|||
|
|
"""User response schema."""
|
|||
|
|
is_active: bool
|
|||
|
|
is_superuser: Optional[bool] = Field(default=False, description="是否为超级管理员")
|
|||
|
|
|
|||
|
|
model_config = {
|
|||
|
|
'from_attributes': True
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def model_validate(cls, obj, *, from_attributes=False):
|
|||
|
|
"""从对象创建响应模型，正确处理is_superuser方法"""
|
|||
|
|
if hasattr(obj, '__dict__'):
|
|||
|
|
data = obj.__dict__.copy()
|
|||
|
|
# 调用is_superuser方法获取布尔值
|
|||
|
|
if hasattr(obj, 'is_admin'):
|
|||
|
|
# 使用同步的 is_admin 属性代替异步的 is_superuser 方法
|
|||
|
|
data['is_superuser'] = obj.is_admin
|
|||
|
|
elif hasattr(obj, 'is_superuser') and not callable(obj.is_superuser):
|
|||
|
|
# 如果is_superuser是属性而不是方法
|
|||
|
|
data['is_superuser'] = obj.is_superuser
|
|||
|
|
return super().model_validate(data)
|
|||
|
|
return super().model_validate(obj, from_attributes=from_attributes)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LoginResponse(BaseModel):
|
|||
|
|
"""登录响应模型，包含令牌和用户信息"""
|
|||
|
|
access_token: str
|
|||
|
|
token_type: str
|
|||
|
|
expires_in: int
|
|||
|
|
user: UserResponse
|