79 lines
3.4 KiB
Python
79 lines
3.4 KiB
Python
from pydantic import BaseModel, Field, field_validator
|
||
from router.common_model import BaseResponse
|
||
from typing import Optional, Dict, Any, Union
|
||
import json
|
||
|
||
|
||
class UserSettings(BaseModel):
|
||
"""用户设置模型"""
|
||
theme: str = Field(..., description="主题模式")
|
||
lightModeOpacity: float = Field(..., description="浅色模式透明度", ge=0.0, le=1.0)
|
||
lightModeBrightness: float = Field(..., description="浅色模式亮度", ge=0.0, le=1.0)
|
||
darkModeOpacity: float = Field(..., description="深色模式透明度", ge=0.0, le=1.0)
|
||
darkModeBrightness: float = Field(..., description="深色模式亮度", ge=0.0, le=1.0)
|
||
backgroundBlur: float = Field(..., description="背景模糊强度", ge=0.0, le=1.0)
|
||
|
||
@field_validator('theme')
|
||
def validate_theme(cls, v):
|
||
"""验证主题值"""
|
||
valid_themes = ['light', 'dark', 'system', 'ThemeMode.light', 'ThemeMode.dark', 'ThemeMode.system']
|
||
if v not in valid_themes:
|
||
raise ValueError(f"无效的主题值: {v}，有效值: {valid_themes}")
|
||
return v
|
||
|
||
|
||
class UserProfileData(BaseModel):
|
||
"""用户资料数据模型"""
|
||
userid: str = Field(..., description="用户ID")
|
||
avatar: Optional[str] = Field(None, description="用户头像base64数据")
|
||
background: Optional[str] = Field(None, description="用户背景base64数据")
|
||
nickname: Optional[str] = Field(None, description="用户昵称")
|
||
settings: Optional[UserSettings] = Field(None, description="用户设置对象")
|
||
|
||
|
||
class GetUserProfileRequest(BaseModel):
|
||
"""获取用户资料请求模型"""
|
||
token: str = Field(..., description="用户认证token")
|
||
|
||
|
||
class UpdateUserProfileRequest(BaseModel):
|
||
"""更新用户资料请求模型"""
|
||
token: str = Field(..., description="用户认证token")
|
||
avatar: Optional[str] = Field(None, description="用户头像base64编码数据")
|
||
background: Optional[str] = Field(None, description="用户背景base64编码数据")
|
||
nickname: Optional[str] = Field(None, description="用户昵称")
|
||
settings: Optional[Union[UserSettings, str]] = Field(None, description="用户设置对象或JSON字符串")
|
||
|
||
@field_validator('settings')
|
||
def parse_settings(cls, v):
|
||
"""解析settings字段，支持字符串和对象两种格式"""
|
||
if v is None:
|
||
return v
|
||
|
||
# 如果已经是UserSettings对象，直接返回
|
||
if isinstance(v, UserSettings):
|
||
return v
|
||
|
||
# 如果是字符串，尝试解析为JSON然后创建UserSettings对象
|
||
if isinstance(v, str):
|
||
try:
|
||
settings_dict = json.loads(v)
|
||
return UserSettings(**settings_dict)
|
||
except json.JSONDecodeError as e:
|
||
raise ValueError(f"settings字段JSON格式错误: {str(e)}")
|
||
except Exception as e:
|
||
raise ValueError(f"settings字段验证失败: {str(e)}")
|
||
|
||
# 如果是字典，直接创建UserSettings对象
|
||
if isinstance(v, dict):
|
||
try:
|
||
return UserSettings(**v)
|
||
except Exception as e:
|
||
raise ValueError(f"settings字段验证失败: {str(e)}")
|
||
|
||
raise ValueError("settings字段必须是JSON字符串、字典或UserSettings对象")
|
||
|
||
|
||
class UserProfileResponse(BaseResponse[UserProfileData]):
|
||
"""用户资料响应模型"""
|
||
pass |