MuMuAINovel/backend/app/schemas/settings.py

"""设置相关的Pydantic模型"""
from pydantic import BaseModel, Field, ConfigDict
from typing import Optional, List
from datetime import datetime


class SettingsBase(BaseModel):
    """设置基础模型"""
    model_config = ConfigDict(protected_namespaces=())
    
    api_provider: Optional[str] = Field(default="openai", description="API提供商")
    api_key: Optional[str] = Field(default=None, description="API密钥")
    api_base_url: Optional[str] = Field(default=None, description="自定义API地址")
    llm_model: Optional[str] = Field(default="gpt-4", description="模型名称")
    temperature: Optional[float] = Field(default=0.7, ge=0.0, le=2.0, description="温度参数")
    max_tokens: Optional[int] = Field(default=2000, ge=1, description="最大token数")
    system_prompt: Optional[str] = Field(default=None, description="系统级别提示词，每次AI调用都会使用")
    cover_api_provider: Optional[str] = Field(default=None, description="封面图片API提供商")
    cover_api_key: Optional[str] = Field(default=None, description="封面图片API密钥")
    cover_api_base_url: Optional[str] = Field(default=None, description="封面图片自定义API地址")
    cover_image_model: Optional[str] = Field(default=None, description="封面图片模型名称")
    cover_enabled: Optional[bool] = Field(default=False, description="是否启用封面图片生成")
    preferences: Optional[str] = Field(default=None, description="其他偏好设置(JSON)")


class SettingsCreate(SettingsBase):
    """创建设置请求模型"""
    pass


class SettingsUpdate(SettingsBase):
    """更新设置请求模型"""
    pass


class SettingsResponse(SettingsBase):
    """设置响应模型"""
    model_config = ConfigDict(from_attributes=True, protected_namespaces=())
    
    id: str
    user_id: str
    created_at: datetime
    updated_at: datetime


class SystemSMTPSettingsBase(BaseModel):
    """系统 SMTP 设置基础模型"""
    model_config = ConfigDict(protected_namespaces=())

    smtp_provider: str = Field(default="qq", description="SMTP 提供商")
    smtp_host: Optional[str] = Field(default=None, description="SMTP 主机")
    smtp_port: int = Field(default=465, ge=1, le=65535, description="SMTP 端口")
    smtp_username: Optional[str] = Field(default=None, description="SMTP 用户名")
    smtp_password: Optional[str] = Field(default=None, description="SMTP 密码或授权码")
    smtp_use_tls: bool = Field(default=False, description="是否启用 TLS")
    smtp_use_ssl: bool = Field(default=True, description="是否启用 SSL")
    smtp_from_email: Optional[str] = Field(default=None, description="发件人邮箱")
    smtp_from_name: str = Field(default="墨木灵思", description="发件人名称")
    email_auth_enabled: bool = Field(default=True, description="是否启用邮箱认证")
    email_register_enabled: bool = Field(default=True, description="是否启用邮箱注册")
    verification_code_ttl_minutes: int = Field(default=10, ge=1, le=120, description="验证码有效期（分钟）")
    verification_resend_interval_seconds: int = Field(default=60, ge=10, le=3600, description="验证码重发间隔（秒）")


class SystemSMTPSettingsUpdate(SystemSMTPSettingsBase):
    """系统 SMTP 设置更新模型"""
    pass


class SystemSMTPSettingsResponse(SystemSMTPSettingsBase):
    """系统 SMTP 设置响应模型"""
    model_config = ConfigDict(from_attributes=True, protected_namespaces=())

    id: str
    user_id: str
    created_at: datetime
    updated_at: datetime


class SMTPTestRequest(BaseModel):
    """SMTP 测试请求模型"""
    model_config = ConfigDict(protected_namespaces=())

    to_email: str = Field(..., min_length=3, max_length=255, description="测试收件邮箱")


# ========== API配置预设相关模型 ==========

class APIKeyPresetConfig(BaseModel):
    """预设配置内容"""
    model_config = ConfigDict(protected_namespaces=())
    
    api_provider: str = Field(..., description="API提供商")
    api_key: str = Field(..., description="API密钥")
    api_base_url: Optional[str] = Field(None, description="自定义API地址")
    llm_model: str = Field(..., description="模型名称")
    temperature: float = Field(default=0.7, ge=0.0, le=2.0, description="温度参数")
    max_tokens: int = Field(default=2000, ge=1, description="最大token数")
    system_prompt: Optional[str] = Field(default=None, description="系统级别提示词")


class APIKeyPreset(BaseModel):
    """API配置预设"""
    model_config = ConfigDict(protected_namespaces=())
    
    id: str = Field(..., description="预设ID")
    name: str = Field(..., min_length=1, max_length=50, description="预设名称")
    description: Optional[str] = Field(None, max_length=200, description="预设描述")
    is_active: bool = Field(default=False, description="是否激活")
    created_at: datetime = Field(..., description="创建时间")
    config: APIKeyPresetConfig = Field(..., description="配置内容")


class PresetCreateRequest(BaseModel):
    """创建预设请求"""
    model_config = ConfigDict(protected_namespaces=())
    
    name: str = Field(..., min_length=1, max_length=50, description="预设名称")
    description: Optional[str] = Field(None, max_length=200, description="预设描述")
    config: APIKeyPresetConfig = Field(..., description="配置内容")


class PresetUpdateRequest(BaseModel):
    """更新预设请求"""
    model_config = ConfigDict(protected_namespaces=())
    
    name: Optional[str] = Field(None, min_length=1, max_length=50, description="预设名称")
    description: Optional[str] = Field(None, max_length=200, description="预设描述")
    config: Optional[APIKeyPresetConfig] = Field(None, description="配置内容")


class PresetResponse(APIKeyPreset):
    """预设响应"""
    pass


class PresetListResponse(BaseModel):
    """预设列表响应"""
    model_config = ConfigDict(protected_namespaces=())
    
    presets: List[PresetResponse] = Field(..., description="预设列表")
    total: int = Field(..., description="总数")
    active_preset_id: Optional[str] = Field(None, description="当前激活的预设ID")
    chapter_analysis_preset_id: Optional[str] = Field(None, description="章节内容分析使用的预设ID，为空则使用默认API配置")


class ChapterAnalysisPresetSelectionRequest(BaseModel):
    """章节内容分析预设选择请求"""
    model_config = ConfigDict(protected_namespaces=())

    preset_id: Optional[str] = Field(None, description="章节内容分析使用的预设ID；为空则使用默认API配置")