LoveACE-EndF/config/models.py

from typing import List, Dict, Any, Optional
from pydantic import BaseModel, Field, field_validator
from enum import Enum


class LogLevel(str, Enum):
    """日志级别枚举"""
    TRACE = "TRACE"
    DEBUG = "DEBUG"
    INFO = "INFO"
    SUCCESS = "SUCCESS"
    WARNING = "WARNING"
    ERROR = "ERROR"
    CRITICAL = "CRITICAL"


class DatabaseConfig(BaseModel):
    """数据库配置"""
    url: str = Field(
        default="mysql+aiomysql://root:123456@localhost:3306/loveac",
        description="数据库连接URL"
    )
    echo: bool = Field(default=False, description="是否启用SQL日志")
    pool_size: int = Field(default=10, description="连接池大小")
    max_overflow: int = Field(default=20, description="连接池最大溢出")
    pool_timeout: int = Field(default=30, description="连接池超时时间(秒)")
    pool_recycle: int = Field(default=3600, description="连接回收时间(秒)")


class AUFEConfig(BaseModel):
    """AUFE连接配置"""
    default_timeout: int = Field(default=30, description="默认超时时间(秒)")
    max_retries: int = Field(default=3, description="最大重试次数")
    max_reconnect_retries: int = Field(default=2, description="最大重连次数")
    activity_timeout: int = Field(default=300, description="活动超时时间(秒)")
    monitor_interval: int = Field(default=60, description="监控间隔(秒)")
    retry_base_delay: float = Field(default=1.0, description="重试基础延迟(秒)")
    retry_max_delay: float = Field(default=60.0, description="重试最大延迟(秒)")
    retry_exponential_base: float = Field(default=2, description="重试指数基数")
    
    # UAAP配置
    uaap_base_url: str = Field(
        default="http://uaap-aufe-edu-cn.vpn2.aufe.edu.cn:8118/cas",
        description="UAAP基础URL"
    )
    uaap_login_url: str = Field(
        default="http://uaap-aufe-edu-cn.vpn2.aufe.edu.cn:8118/cas/login?service=http%3A%2F%2Fjwcxk2.aufe.edu.cn%2Fj_spring_cas_security_check",
        description="UAAP登录URL"
    )
    
    # 默认请求头
    default_headers: Dict[str, str] = Field(
        default_factory=lambda: {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
        },
        description="默认请求头"
    )


class S3Config(BaseModel):
    """S3客户端配置"""
    access_key_id: str = Field(default="", description="S3访问密钥ID")
    secret_access_key: str = Field(default="", description="S3秘密访问密钥")
    endpoint_url: Optional[str] = Field(default=None, description="S3终端节点URL")
    region_name: str = Field(default="us-east-1", description="S3区域名称")
    bucket_name: str = Field(default="", description="默认存储桶名称")
    use_ssl: bool = Field(default=True, description="是否使用SSL")
    signature_version: str = Field(default="s3v4", description="签名版本")
    
    @field_validator('access_key_id', 'secret_access_key', 'bucket_name')
    @classmethod
    def validate_required_fields(cls, v):
        """验证必填字段"""
        # 允许为空，但应在运行时检查
        return v


class LogConfig(BaseModel):
    """日志配置"""
    level: LogLevel = Field(default=LogLevel.INFO, description="日志级别")
    format: str = Field(
        default="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
        description="日志格式"
    )
    file_path: str = Field(default="logs/app.log", description="日志文件路径")
    rotation: str = Field(default="10 MB", description="日志轮转大小")
    retention: str = Field(default="30 days", description="日志保留时间")
    compression: str = Field(default="zip", description="日志压缩格式")
    backtrace: bool = Field(default=True, description="是否启用回溯")
    diagnose: bool = Field(default=True, description="是否启用诊断")
    console_output: bool = Field(default=True, description="是否输出到控制台")
    
    # 额外的日志文件配置
    additional_loggers: List[Dict[str, Any]] = Field(
        default_factory=lambda: [
            {
                "file_path": "logs/debug.log",
                "level": "DEBUG",
                "rotation": "10 MB"
            },
            {
                "file_path": "logs/error.log", 
                "level": "ERROR",
                "rotation": "10 MB"
            }
        ],
        description="额外的日志记录器配置"
    )


class AppConfig(BaseModel):
    """应用程序配置"""
    title: str = Field(default="LoveAC API", description="应用标题")
    description: str = Field(default="LoveACAPI API", description="应用描述")
    version: str = Field(default="1.0.0", description="应用版本")
    debug: bool = Field(default=False, description="是否启用调试模式")
    
    # CORS配置
    cors_allow_origins: List[str] = Field(
        default_factory=lambda: ["*"],
        description="允许的CORS来源"
    )
    cors_allow_credentials: bool = Field(default=True, description="是否允许CORS凭据")
    cors_allow_methods: List[str] = Field(
        default_factory=lambda: ["*"],
        description="允许的CORS方法"
    )
    cors_allow_headers: List[str] = Field(
        default_factory=lambda: ["*"],
        description="允许的CORS头部"
    )
    
    # 服务器配置
    host: str = Field(default="0.0.0.0", description="服务器主机")
    port: int = Field(default=8000, description="服务器端口")
    workers: int = Field(default=1, description="工作进程数")


class Settings(BaseModel):
    """主配置类"""
    database: DatabaseConfig = Field(default_factory=DatabaseConfig)
    aufe: AUFEConfig = Field(default_factory=AUFEConfig)
    s3: S3Config = Field(default_factory=S3Config)
    log: LogConfig = Field(default_factory=LogConfig)
    app: AppConfig = Field(default_factory=AppConfig)
    
    class Config:
        json_encoders = {
            # 为枚举类型提供JSON编码器
            LogLevel: lambda v: v.value
        }