config.py

import yaml
from typing import Dict, Any, Optional

class LLMConfig:
    def __init__(self, config: Dict[str, Any]):
        self.config = config
        self.name = config["name"]
        self.key = config["key"]
        self.model = config["model"]
        self.base_url = config["base_url"]


    def get_llm_config(self) -> Dict[str, Any]:
        """获取 LLM 配置"""
        return self.config.get("llms", {})
    
    def get_llm_by_name(self, name: str) -> Optional[Dict[str, Any]]:
        """根据名称获取特定的LLM配置"""
        llms = self.get_llm_config()
        for llm in llms:
            if llm.get("name") == name:
                return llm
        return None

    def get_key(self) -> Optional[str]:
        """根据名称获取特定的LLM配置的key"""
        return self.key
    
    def get_model(self) -> Optional[str]:
        """根据名称获取特定的LLM配置的model"""
        return self.model
    
    def get_base_url(self) -> Optional[str]:
        """根据名称获取特定的LLM配置的base_url"""
        return self.base_url

class ConfigManager:
    """
    YAML 配置文件管理类，用于加载、解析和访问配置信息
    """
    
    def __init__(self, config_path: str = "confs/config.yaml"):
        """
        初始化配置管理器
        
        Args:
            config_path: YAML 配置文件的路径
        """
        self.config_path = config_path
        self.config = self._load_config()
        llms = self.get_llm_config()
        self.llms = {}
        for llm in llms:
            self.llms[llm.get("name")] = LLMConfig(llm)
    
    def _load_config(self) -> Dict[str, Any]:
        """
        加载并解析 YAML 配置文件
        
        Returns:
            解析后的配置字典
        """
        try:
            with open(self.config_path, 'r', encoding='utf-8') as file:
                return yaml.safe_load(file)
        except Exception as e:
            print(f"加载配置文件时出错: {e}")
            return {}
    
    def reload(self) -> None:
        """重新加载配置文件"""
        self.config = self._load_config()
    
    def get_config(self) -> Dict[str, Any]:
        """获取完整配置"""
        return self.config
    
    def get_llm_config(self) -> Dict[str, Any]:
        """获取 LLM 配置列表"""
        return self.config.get("llms", [])
    
    def get_llm_by_name(self, name: str) -> Optional[Dict[str, Any]]:
        """根据名称获取特定的LLM配置"""
        if self.llms is None or self.llms.get(name) is None:
            return None
        return self.llms.get(name)
    
    def get_translate_config(self) -> Dict[str, Any]:
        """获取翻译配置"""
        return self.config.get("translate", {})
    
    def get_translate_model(self) -> Optional[str]:
        """获取翻译模型"""
        return self.get_translate_config().get("model")
    
    def get_translate_languages(self) -> tuple:
        """获取翻译语言对 (源语言, 目标语言)"""
        config = self.get_translate_config()
        return (
            config.get("source_language"),
            config.get("target_language")
        )
    
    def get_subtitles_config(self) -> Dict[str, Any]:
        """获取字幕配置"""
        return self.config.get("subtitles", {})
    
    def get_subtitles_model(self) -> Optional[str]:
        """获取字幕模型"""
        return self.get_subtitles_config().get("model")
    
    def get_subtitles_languages(self) -> tuple:
        """获取字幕语言对 (源语言, 目标语言)"""
        config = self.get_subtitles_config()
        return (
            config.get("source_language"),
            config.get("target_language")
        )
    
    def save_config(self, config: Dict[str, Any] = None) -> bool:
        """
        保存配置到文件
        
        Args:
            config: 要保存的配置，默认为当前配置
            
        Returns:
            是否保存成功
        """
        if config is None:
            config = self.config
            
        try:
            with open(self.config_path, 'w', encoding='utf-8') as file:
                yaml.dump(config, file, default_flow_style=False, allow_unicode=True)
            return True
        except Exception as e:
            print(f"保存配置文件时出错: {e}")
            return False


if __name__ == "__main__":
    # 使用示例
    config_manager = ConfigManager()
    
    # 获取所有LLM配置
    llms = config_manager.get_llm_config()
    print(f"配置的LLM模型数量: {len(llms)}")
    
    
    # 获取翻译配置
    translate_model = config_manager.get_translate_model()
    source_lang, target_lang = config_manager.get_translate_languages()
    print(f"\n翻译模型: {translate_model}")
    print(f"翻译语言: {source_lang} -> {target_lang}")
    
    # 获取字幕配置
    subtitles_model = config_manager.get_subtitles_model()
    sub_source_lang, sub_target_lang = config_manager.get_subtitles_languages()
    print(f"\n字幕模型: {subtitles_model}")
    print(f"字幕语言: {sub_source_lang} -> {sub_target_lang}")