K2cr2O1
ed4da64a7a
- 在requirements.txt中添加aiomysql依赖 - 在config.toml中添加MySQL配置块 - 新增MySQLModel配置模型 - 实现MySQLManager单例管理器 - 更新Config类以支持MySQL配置加载 - 在__init__.py中导出mysql_manager - 改进ConfigError异常处理
84 lines
1.9 KiB
Python
84 lines
1.9 KiB
Python
"""
|
|
Pydantic 配置模型模块
|
|
|
|
该模块使用 Pydantic 定义了与 `config.toml` 文件结构完全对应的配置模型。
|
|
这使得配置的加载、校验和访问都变得类型安全和健壮。
|
|
"""
|
|
from typing import List, Optional
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class NapCatWSModel(BaseModel):
|
|
"""
|
|
对应 `config.toml` 中的 `[napcat_ws]` 配置块。
|
|
"""
|
|
uri: str
|
|
token: str
|
|
reconnect_interval: int = 5
|
|
|
|
|
|
class BotModel(BaseModel):
|
|
"""
|
|
对应 `config.toml` 中的 `[bot]` 配置块。
|
|
"""
|
|
command: List[str] = Field(default_factory=lambda: ["/"])
|
|
ignore_self_message: bool = True
|
|
permission_denied_message: str = "权限不足，需要 {permission_name} 权限"
|
|
|
|
|
|
class RedisModel(BaseModel):
|
|
"""
|
|
对应 `config.toml` 中的 `[redis]` 配置块。
|
|
"""
|
|
host: str
|
|
port: int
|
|
db: int
|
|
password: str
|
|
|
|
|
|
class MySQLModel(BaseModel):
|
|
"""
|
|
对应 `config.toml` 中的 `[mysql]` 配置块。
|
|
"""
|
|
host: str
|
|
port: int
|
|
user: str
|
|
password: str
|
|
db: str
|
|
charset: str = "utf8mb4"
|
|
|
|
|
|
class DockerModel(BaseModel):
|
|
"""
|
|
对应 `config.toml` 中的 `[docker]` 配置块。
|
|
"""
|
|
base_url: Optional[str] = None
|
|
sandbox_image: str = "python-sandbox:latest"
|
|
timeout: int = 10
|
|
concurrency_limit: int = 5
|
|
tls_verify: bool = False
|
|
ca_cert_path: Optional[str] = None
|
|
client_cert_path: Optional[str] = None
|
|
client_key_path: Optional[str] = None
|
|
|
|
class ImageManagerModel(BaseModel):
|
|
"""
|
|
对应 `config.toml` 中的 `[image_manager]` 配置块。
|
|
"""
|
|
image_height: int = 1920
|
|
image_width: int = 1080
|
|
|
|
|
|
class ConfigModel(BaseModel):
|
|
"""
|
|
顶层配置模型，整合了所有子配置块。
|
|
"""
|
|
napcat_ws: NapCatWSModel
|
|
bot: BotModel
|
|
redis: RedisModel
|
|
mysql: MySQLModel
|
|
docker: DockerModel
|
|
image_manager: ImageManagerModel
|
|
|
|
|