🎉初次提交

config/manager.py

@@ -0,0 +1,177 @@
+import json
+import os
+from pathlib import Path
+from typing import Any, Dict, Optional
+from loguru import logger
+from pydantic import ValidationError
+
+from .models import Settings
+
+
+class ConfigManager:
+    """配置文件管理器"""
+    
+    def __init__(self, config_file: str = "config.json"):
+        self.config_file = Path(config_file)
+        self._settings: Optional[Settings] = None
+        self._ensure_config_dir()
+    
+    def _ensure_config_dir(self):
+        """确保配置文件目录存在"""
+        self.config_file.parent.mkdir(parents=True, exist_ok=True)
+    
+    def _create_default_config(self) -> Settings:
+        """创建默认配置"""
+        logger.info("正在创建默认配置文件...")
+        return Settings()
+    
+    def _save_config(self, settings: Settings):
+        """保存配置到文件"""
+        try:
+            config_dict = settings.dict()
+            with open(self.config_file, 'w', encoding='utf-8') as f:
+                json.dump(config_dict, f, indent=2, ensure_ascii=False)
+            logger.info(f"配置已保存到 {self.config_file}")
+        except Exception as e:
+            logger.error(f"保存配置文件失败: {e}")
+            raise
+    
+    def _load_config(self) -> Settings:
+        """从文件加载配置"""
+        if not self.config_file.exists():
+            logger.warning(f"配置文件 {self.config_file} 不存在，将创建默认配置")
+            settings = self._create_default_config()
+            self._save_config(settings)
+            return settings
+        
+        try:
+            with open(self.config_file, 'r', encoding='utf-8') as f:
+                config_data = json.load(f)
+            
+            # 验证并创建Settings对象
+            settings = Settings(**config_data)
+            logger.info(f"成功加载配置文件: {self.config_file}")
+            return settings
+            
+        except json.JSONDecodeError as e:
+            logger.error(f"配置文件JSON格式错误: {e}")
+            raise
+        except ValidationError as e:
+            logger.error(f"配置文件验证失败: {e}")
+            raise
+        except Exception as e:
+            logger.error(f"加载配置文件失败: {e}")
+            raise
+    
+    def get_settings(self) -> Settings:
+        """获取配置设置"""
+        if self._settings is None:
+            self._settings = self._load_config()
+        return self._settings
+    
+    def reload_config(self) -> Settings:
+        """重新加载配置"""
+        logger.info("正在重新加载配置...")
+        self._settings = self._load_config()
+        return self._settings
+    
+    def update_config(self, **kwargs) -> Settings:
+        """更新配置"""
+        settings = self.get_settings()
+        
+        # 创建新的配置字典
+        config_dict = settings.dict()
+        
+        # 更新指定的配置项
+        for key, value in kwargs.items():
+            if '.' in key:
+                # 支持嵌套键，如 'database.url'
+                keys = key.split('.')
+                current = config_dict
+                for k in keys[:-1]:
+                    if k not in current:
+                        current[k] = {}
+                    current = current[k]
+                current[keys[-1]] = value
+            else:
+                config_dict[key] = value
+        
+        try:
+            # 验证更新后的配置
+            new_settings = Settings(**config_dict)
+            self._save_config(new_settings)
+            self._settings = new_settings
+            logger.info("配置更新成功")
+            return new_settings
+        except ValidationError as e:
+            logger.error(f"配置更新失败，验证错误: {e}")
+            raise
+    
+    def validate_config(self) -> bool:
+        """验证配置完整性"""
+        try:
+            settings = self.get_settings()
+            
+            # 检查关键配置项
+            issues = []
+            
+            # 检查数据库配置
+            if not settings.database.url:
+                issues.append("数据库URL未配置")
+            
+            # 检查S3配置（如果需要使用）
+            if settings.s3.bucket_name and not settings.s3.access_key_id:
+                issues.append("S3配置不完整：缺少access_key_id")
+            if settings.s3.bucket_name and not settings.s3.secret_access_key:
+                issues.append("S3配置不完整：缺少secret_access_key")
+            
+            # 检查日志配置
+            log_dir = Path(settings.log.file_path).parent
+            if not log_dir.exists():
+                try:
+                    log_dir.mkdir(parents=True, exist_ok=True)
+                    logger.info(f"创建日志目录: {log_dir}")
+                except Exception as e:
+                    issues.append(f"无法创建日志目录 {log_dir}: {e}")
+            
+            if issues:
+                logger.warning("配置验证发现问题:")
+                for issue in issues:
+                    logger.warning(f"  - {issue}")
+                return False
+            
+            logger.info("配置验证通过")
+            return True
+            
+        except Exception as e:
+            logger.error(f"配置验证失败: {e}")
+            return False
+    
+    def get_config_summary(self) -> Dict[str, Any]:
+        """获取配置摘要（隐藏敏感信息）"""
+        settings = self.get_settings()
+        config_dict = settings.dict()
+        
+        # 隐藏敏感信息
+        sensitive_keys = [
+            'database.url',
+            's3.access_key_id',
+            's3.secret_access_key'
+        ]
+        
+        def hide_sensitive(data: Dict[str, Any], keys: list, prefix: str = ""):
+            for key, value in data.items():
+                current_key = f"{prefix}.{key}" if prefix else key
+                if current_key in sensitive_keys:
+                    if isinstance(value, str) and value:
+                        data[key] = value[:8] + "..." if len(value) > 8 else "***"
+                elif isinstance(value, dict):
+                    hide_sensitive(value, keys, current_key)
+        
+        summary = config_dict.copy()
+        hide_sensitive(summary, sensitive_keys)
+        return summary
+
+
+# 全局配置管理器实例
+config_manager = ConfigManager()