LiteyukiBot/liteyuki/utils/config.py

65 lines
2.0 KiB
Python
Raw Normal View History

2024-03-19 05:16:25 +00:00
import os
import nonebot
import yaml
from pydantic import BaseModel
2024-03-18 10:21:56 +00:00
2024-04-11 05:15:29 +00:00
from liteyuki.utils.data_manager import StoredConfig, common_db
from liteyuki.utils.ly_typing import T_Bot
from liteyuki.utils.tools import random_hex_string
config = {} # 全局配置,确保加载后读取
2024-03-18 16:27:40 +00:00
2024-03-18 10:21:56 +00:00
2024-03-19 05:16:25 +00:00
class BasicConfig(BaseModel):
host: str = "127.0.0.1"
port: int = 20216
superusers: list[str] = []
command_start: list[str] = ["/", ""]
nickname: list[str] = [f"LiteyukiBot-{random_hex_string(6)}"]
2024-03-19 05:16:25 +00:00
2024-03-18 10:21:56 +00:00
def load_from_yaml(file: str) -> dict:
2024-03-18 16:27:40 +00:00
global config
2024-03-24 13:21:57 +00:00
nonebot.logger.debug("Loading config from %s" % file)
2024-03-19 05:16:25 +00:00
if not os.path.exists(file):
2024-03-26 09:14:41 +00:00
nonebot.logger.warning(f"Config file {file} not found, created default config, please modify it and restart")
with open(file, "w", encoding="utf-8") as f:
2024-03-19 05:16:25 +00:00
yaml.dump(BasicConfig().dict(), f, default_flow_style=False)
2024-03-26 09:14:41 +00:00
with open(file, "r", encoding="utf-8") as f:
conf = init_conf(yaml.load(f, Loader=yaml.FullLoader))
2024-03-19 05:16:25 +00:00
config = conf
if conf is None:
2024-03-26 09:14:41 +00:00
nonebot.logger.warning(f"Config file {file} is empty, use default config. please modify it and restart")
2024-03-19 05:16:25 +00:00
conf = BasicConfig().dict()
return conf
2024-04-11 05:15:29 +00:00
def get_config(key: str, bot: T_Bot = None, default=None):
"""获取配置项优先级bot > config > db > yaml"""
if bot is None:
bot_config = {}
else:
bot_config = bot.config.dict()
if key in bot_config:
return bot_config[key]
elif key in config:
return config[key]
elif key in common_db.first(StoredConfig(), default=StoredConfig()).config:
return common_db.first(StoredConfig(), default=StoredConfig()).config[key]
elif key in load_from_yaml("config.yml"):
return load_from_yaml("config.yml")[key]
else:
return default
def init_conf(conf: dict) -> dict:
if "" not in conf.get("command_start", []):
conf["alconna_use_command_start"] = True
return conf