K2cr2O1
ed4da64a7a
- 在requirements.txt中添加aiomysql依赖 - 在config.toml中添加MySQL配置块 - 新增MySQLModel配置模型 - 实现MySQLManager单例管理器 - 更新Config类以支持MySQL配置加载 - 在__init__.py中导出mysql_manager - 改进ConfigError异常处理
135 lines
4.1 KiB
Python
135 lines
4.1 KiB
Python
"""
|
|
配置加载模块
|
|
|
|
负责读取和解析 config.toml 配置文件，提供全局配置对象。
|
|
"""
|
|
from pathlib import Path
|
|
|
|
import tomllib
|
|
from pydantic import ValidationError
|
|
from .config_models import ConfigModel, NapCatWSModel, BotModel, RedisModel, DockerModel, ImageManagerModel, MySQLModel
|
|
from .utils.logger import ModuleLogger
|
|
from .utils.exceptions import ConfigError, ConfigNotFoundError, ConfigValidationError
|
|
|
|
|
|
class Config:
|
|
"""
|
|
配置加载类，负责读取和解析 config.toml 文件
|
|
"""
|
|
|
|
def __init__(self, file_path: str = "config.toml"):
|
|
"""
|
|
初始化配置加载器
|
|
|
|
:param file_path: 配置文件路径，默认为 "config.toml"
|
|
"""
|
|
self.path = Path(file_path)
|
|
self._model: ConfigModel
|
|
# 创建模块专用日志记录器
|
|
self.logger = ModuleLogger("ConfigLoader")
|
|
self.load()
|
|
|
|
def load(self):
|
|
"""
|
|
加载并验证配置文件
|
|
|
|
:raises ConfigNotFoundError: 如果配置文件不存在
|
|
:raises ConfigValidationError: 如果配置格式不正确
|
|
:raises ConfigError: 如果加载配置时发生其他错误
|
|
"""
|
|
if not self.path.exists():
|
|
error = ConfigNotFoundError(message=f"配置文件 {self.path} 未找到！")
|
|
self.logger.error(f"配置加载失败: {error.message}")
|
|
self.logger.log_custom_exception(error)
|
|
raise error
|
|
|
|
try:
|
|
self.logger.info(f"正在从 {self.path} 加载配置...")
|
|
with open(self.path, "rb") as f:
|
|
raw_config = tomllib.load(f)
|
|
|
|
self._model = ConfigModel(**raw_config)
|
|
self.logger.success("配置加载并验证成功！")
|
|
|
|
except ValidationError as e:
|
|
error_details = []
|
|
for error in e.errors():
|
|
field = " -> ".join(map(str, error["loc"]))
|
|
error_msg = f"字段 '{field}': {error['msg']}"
|
|
error_details.append(error_msg)
|
|
|
|
validation_error = ConfigValidationError(
|
|
message="配置验证失败"
|
|
)
|
|
validation_error.original_error = e
|
|
|
|
self.logger.error("配置验证失败，请检查 `config.toml` 文件中的以下错误：")
|
|
for detail in error_details:
|
|
self.logger.error(f" - {detail}")
|
|
|
|
self.logger.log_custom_exception(validation_error)
|
|
raise validation_error
|
|
except tomllib.TOMLDecodeError as e:
|
|
error = ConfigError(
|
|
message=f"TOML解析错误: {str(e)}"
|
|
)
|
|
error.original_error = e
|
|
self.logger.error(f"加载配置文件时发生TOML解析错误: {error.message}")
|
|
self.logger.log_custom_exception(error)
|
|
raise error
|
|
except Exception as e:
|
|
error = ConfigError(
|
|
message=f"加载配置文件时发生未知错误: {str(e)}"
|
|
)
|
|
error.original_error = e
|
|
self.logger.exception(f"加载配置文件时发生未知错误: {error.message}")
|
|
self.logger.log_custom_exception(error)
|
|
raise error
|
|
|
|
# 通过属性访问配置
|
|
@property
|
|
def napcat_ws(self) -> NapCatWSModel:
|
|
"""
|
|
获取 NapCat WebSocket 配置
|
|
"""
|
|
return self._model.napcat_ws
|
|
|
|
@property
|
|
def bot(self) -> BotModel:
|
|
"""
|
|
获取 Bot 基础配置
|
|
"""
|
|
return self._model.bot
|
|
|
|
@property
|
|
def redis(self) -> RedisModel:
|
|
"""
|
|
获取 Redis 配置
|
|
"""
|
|
return self._model.redis
|
|
|
|
@property
|
|
def mysql(self) -> MySQLModel:
|
|
"""
|
|
获取 MySQL 配置
|
|
"""
|
|
return self._model.mysql
|
|
|
|
@property
|
|
def docker(self) -> DockerModel:
|
|
"""
|
|
获取 Docker 配置
|
|
"""
|
|
return self._model.docker
|
|
|
|
@property
|
|
def image_manager(self) -> ImageManagerModel:
|
|
"""
|
|
获取图片生成管理器配置
|
|
"""
|
|
return self._model.image_manager
|
|
|
|
|
|
# 实例化全局配置对象
|
|
global_config = Config()
|