63 lines
2.6 KiB
Python
63 lines
2.6 KiB
Python
import os
|
||
import sys
|
||
import tomli
|
||
import shutil
|
||
from .logger import logger
|
||
from typing import Optional
|
||
|
||
|
||
class Config:
|
||
platform: str = "qq"
|
||
nickname: Optional[str] = None
|
||
server_host: str = "localhost"
|
||
server_port: int = 8095
|
||
napcat_heartbeat_interval: int = 30
|
||
|
||
def __init__(self):
|
||
self._get_config_path()
|
||
|
||
def _get_config_path(self):
|
||
current_file_path = os.path.abspath(__file__)
|
||
src_path = os.path.dirname(current_file_path)
|
||
self.root_path = os.path.join(src_path, "..")
|
||
self.config_path = os.path.join(self.root_path, "config.toml")
|
||
|
||
def load_config(self): # sourcery skip: extract-method, move-assign
|
||
include_configs = ["Nickname", "Napcat_Server", "MaiBot_Server", "Debug", "Voice"]
|
||
if os.path.exists(self.config_path):
|
||
with open(self.config_path, "rb") as f:
|
||
try:
|
||
raw_config = tomli.load(f)
|
||
except tomli.TOMLDecodeError as e:
|
||
logger.critical(f"配置文件bot_config.toml填写有误，请检查第{e.lineno}行第{e.colno}处：{e.msg}")
|
||
sys.exit(1)
|
||
for key in include_configs:
|
||
if key not in raw_config:
|
||
logger.error(f"配置文件中缺少必需的字段: '{key}'")
|
||
sys.exit(1)
|
||
self.nickname = raw_config["Nickname"].get("nickname")
|
||
self.server_host = raw_config["Napcat_Server"].get("host", "localhost")
|
||
self.server_port = raw_config["Napcat_Server"].get("port", 8095)
|
||
self.platform = raw_config["MaiBot_Server"].get("platform_name")
|
||
if not self.platform:
|
||
logger.critical("请在配置文件中指定平台")
|
||
sys.exit(1)
|
||
self.napcat_heartbeat_interval = raw_config["Napcat_Server"].get("heartbeat", 30)
|
||
self.mai_host = raw_config["MaiBot_Server"].get("host", "localhost")
|
||
self.mai_port = raw_config["MaiBot_Server"].get("port", 8000)
|
||
self.debug_level = raw_config["Debug"].get("level", "INFO")
|
||
self.use_tts = raw_config["Voice"].get("use_tts", False)
|
||
else:
|
||
logger.error("配置文件不存在！")
|
||
logger.info("正在创建配置文件...")
|
||
shutil.copy(
|
||
os.path.join(self.root_path, "template", "template_config.toml"),
|
||
os.path.join(self.root_path, "config.toml"),
|
||
)
|
||
logger.info("配置文件创建成功，请修改配置文件后重启程序。")
|
||
sys.exit(1)
|
||
|
||
|
||
global_config = Config()
|
||
global_config.load_config()
|