"""统一配置基类 — 所有服务共享的基础配置。 数据库/Redis/OSS/Celery/AI服务等通用配置统一定义在此。 API 和 Worker 各自的 Settings 类继承本类,只追加服务特有字段。 单例模式和 env 文件加载逻辑也统一在这里实现。 """ import os from typing import Optional, TypeVar from pydantic import AliasChoices, Field from pydantic_settings import BaseSettings, SettingsConfigDict T = TypeVar("T", bound=BaseSettings) class SharedSettings(BaseSettings): """所有服务共享的基础配置。 API 和 Worker 都继承本类,确保: 1. 数据库/Redis/OSS/Celery 等核心配置默认值一致 2. 环境变量命名统一(snake_case,pydantic-settings 自动兼容大写) 3. env 文件加载逻辑只实现一次 """ # ── 环境 ────────────────────────────────────────────────────────────── environment: str = "development" debug: bool = True auto_create_schema: bool = False # ── 数据库 ──────────────────────────────────────────────────────────── database_url: str = "postgresql+psycopg://postgres:postgres@localhost:5432/xiaoxia_saas" database_pool_size: int = 20 database_max_overflow: int = 10 # pool_size(20) + max_overflow(10) = 最大30连接 database_pool_timeout: int = 30 database_pool_recycle: int = 3600 # 测试用:使用 SQLite 内存数据库(CI 环境无需 PostgreSQL) use_in_memory_db: bool = False # ── Redis ──────────────────────────────────────────────────────────── redis_url: str = "redis://localhost:6379/0" # ── Celery ─────────────────────────────────────────────────────────── celery_broker_url: str = "redis://localhost:6379/0" celery_result_backend: str = "redis://localhost:6379/1" # ── OSS 阿里云 ────────────────────────────────────────────────────── oss_endpoint: str = "oss-cn-hangzhou.aliyuncs.com" oss_access_key_id: str = "" oss_access_key_secret: str = "" oss_bucket_name: str = "xiaoxia-autocut" oss_direct_upload_max_mb: int = 2000 oss_direct_upload_expire_seconds: int = 900 # ── CosyVoice (阿里云百炼语音合成) ─────────────────────────────────── cosyvoice_api_key: str = "" cosyvoice_base_url: str = "https://dashscope.aliyuncs.com/api/v1" cosyvoice_model: str = "cosyvoice-v3-flash" cosyvoice_voice: str = "longxiaochun_v3" # 默认音色(v3 系列系统音色带 _v3 后缀) cosyvoice_sample_rate: int = 22050 cosyvoice_format: str = "mp3" # 输出格式:mp3/wav/pcm # 音色克隆模型名(固定为 voice-enrollment) cosyvoice_clone_model: str = "voice-enrollment" # ── 豆包大模型(火山引擎方舟) ──────────────────────────────────────── doubao_api_key: str = "" doubao_model: str = "doubao-seed-1-6-250615" doubao_base_url: str = "https://ark.cn-beijing.volces.com/api/v3" doubao_timeout: int = 30 doubao_max_retries: int = 2 doubao_vision_model: str = "doubao-1-5-vision-pro-250915" # ── MediaKit (火山引擎 AI 媒体工具) ────────────────────────────────── mediakit_api_key: str = "" mediakit_base_url: str = "https://mediakit.cn-beijing.volces.com/api/v1" mediakit_timeout: int = 60 # ── 积分/会员系统 (#1895) ──────────────────────────────────────────── # 积分系统总开关(产品要求 #1895:暂停积分系统但保留全部代码/表/接口)。 # - false(默认):所有 AI 功能(生成视频/口型/数字人/AI标题/TTS/克隆音色…) # 对全部登录用户免费放行,不扣积分、不做余额拦截;积分余额/流水/会员 # 状态等查询接口保持可用,但数据不再变动。 # - 未来恢复:只需设置环境变量 ENABLE_CREDIT_SYSTEM=true。 # 旧开关 POINTS_ENABLED 仍保留作为兼容别名(两者任一为 true 即启用)。 # 主开关(推荐环境变量名 ENABLE_CREDIT_SYSTEM) credits_enabled: bool = Field( default=False, validation_alias=AliasChoices("ENABLE_CREDIT_SYSTEM", "credits_enabled"), ) # 旧开关兼容(POINTS_ENABLED);两者任一为 true 即启用 points_enabled_compat: bool = Field( default=False, validation_alias=AliasChoices("POINTS_ENABLED", "points_enabled_compat"), ) @property def points_enabled(self) -> bool: """旧代码/测试使用的属性名,等价于积分系统总开关(兼容别名)。""" return bool(self.credits_enabled or self.points_enabled_compat) @points_enabled.setter def points_enabled(self, value: bool) -> None: # 支持旧测试/代码 ``settings.points_enabled = True`` 的写法 self.credits_enabled = bool(value) self.points_enabled_compat = False # ── GPU MuseTalk 反向轮询 Worker ──────────────────────────────────── # Worker 用这个长期 Token 鉴权(不是用户 JWT)。多 Worker 共用同一个 Token; # worker_id 用于区分具体机器。生产必须配置;development 留空会跳过校验。 gpu_worker_token: str = "" # GPU 任务超时(秒):processing 状态超过此时长(以任务心跳为准)才回退 # pending / failed。#1970:RTX2060 6G 推理 720p 长视频需 5 分钟以上,300→900。 # Worker 推理期间每 30s 通过 /gpu/register(task_id=...) 续心跳, # 只有真正超时或 Worker 明确上报 failed 才会回退。 gpu_task_timeout_seconds: int = 900 # 结果预签名 URL 有效期(秒) gpu_result_url_expires: int = 3600 # 输入预签名 URL 有效期(秒,需留出 Worker 下载时间) gpu_input_url_expires: int = 3600 # 业务侧是否启用 GPU 口型同步(开关);关或无可用 Worker 时回退 MediaKit 云端 use_gpu_lipsync: bool = False # 业务侧轮询 GPU 任务结果的间隔(秒) gpu_lipsync_poll_interval: float = 5.0 # 业务侧等待 GPU 任务结果的总超时(秒);超时后回退 MediaKit。 # 应小于等于 gpu_task_timeout_seconds(默认900s)+ 冗余,留足 Worker 下载/上传时间。 gpu_lipsync_wait_timeout: int = 1200 # 判断 Worker 可用的心跳新鲜度窗口(秒)—— last_heartbeat_at 在窗口内视为在线 gpu_worker_stale_seconds: int = 300 @property def effective_database_url(self) -> str: """返回实际使用的数据库 URL。 当 USE_IN_MEMORY_DB=True 时返回 SQLite 内存 URL,否则返回 database_url。 """ if self.use_in_memory_db: return "sqlite:///./test.db" return self.database_url model_config = SettingsConfigDict( env_file=".env", env_file_encoding="utf-8", case_sensitive=False, extra="ignore", ) # ── 统一单例管理 ──────────────────────────────────────────────────────── # 所有 Settings 类的单例缓存都在这里,消除每处各自实现的重复代码 _settings_cache: dict[str, BaseSettings] = {} def _get_env_file() -> str: """根据 APP_ENV 决定读取哪个 env 文件。""" env = os.getenv("APP_ENV", "development") env_file = f".env.{env}" if env != "development" else ".env" return env_file if os.path.exists(env_file) else ".env" def get_cached_settings(settings_class: type[T], cache_key: Optional[str] = None) -> T: """统一的 Settings 单例获取函数。 所有服务都通过这个函数获取配置,消除重复的单例实现。 按类名缓存,同一类只初始化一次。 """ key = cache_key or settings_class.__name__ if key not in _settings_cache: env_file = _get_env_file() _settings_cache[key] = settings_class(_env_file=env_file) return _settings_cache[key] # type: ignore[return-value] def reload_settings_cache() -> None: """清空配置缓存,下次获取时重新加载。测试用。""" _settings_cache.clear() def get_shared_settings() -> SharedSettings: """获取共享配置单例(统一入口)。""" return get_cached_settings(SharedSettings)