mirror of
https://github.com/TriM-Organization/LiteyukiBot-TriM.git
synced 2025-06-03 03:15:26 +00:00
109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
import os
|
||
import platform
|
||
from typing import List
|
||
|
||
import nonebot
|
||
import yaml
|
||
from pydantic import BaseModel
|
||
|
||
from ..message.tools import random_hex_string
|
||
|
||
|
||
config = {} # 全局配置，确保加载后读取
|
||
|
||
|
||
class SatoriNodeConfig(BaseModel):
|
||
host: str = ""
|
||
port: str = "5500"
|
||
path: str = ""
|
||
token: str = ""
|
||
|
||
|
||
class SatoriConfig(BaseModel):
|
||
comment: str = "此皆正处于开发之中，切勿在生产环境中启用。"
|
||
enable: bool = False
|
||
hosts: List[SatoriNodeConfig] = [SatoriNodeConfig()]
|
||
|
||
|
||
class BasicConfig(BaseModel):
|
||
host: str = "127.0.0.1"
|
||
port: int = 20247
|
||
superusers: list[str] = []
|
||
command_start: list[str] = ["/", ""]
|
||
nickname: list[str] = [f"灵温-{random_hex_string(6)}"]
|
||
default_language: str = "zh-WY"
|
||
satori: SatoriConfig = SatoriConfig()
|
||
data_path: str = "data/liteyuki"
|
||
chromium_path: str = (
|
||
"/Applications/Google\ Chrome.app/Contents/MacOS/Google\ Chrome" # type: ignore
|
||
if platform.system() == "Darwin"
|
||
else (
|
||
"C:/Program Files (x86)/Microsoft/Edge/Application/msedge.exe"
|
||
if platform.system() == "Windows"
|
||
else "/usr/bin/chromium-browser"
|
||
)
|
||
)
|
||
|
||
|
||
def load_from_yaml(file_: str) -> dict:
|
||
global config
|
||
nonebot.logger.debug("正在从 {} 中加载配置项".format(file_))
|
||
if not os.path.exists(file_):
|
||
nonebot.logger.warning(
|
||
f"未寻得配置文件 {file_} ，已以默认配置创建，请在重启后更改为你所需的内容。"
|
||
)
|
||
with open(file_, "w", encoding="utf-8") as f:
|
||
yaml.dump(BasicConfig().dict(), f, default_flow_style=False)
|
||
|
||
with open(file_, "r", encoding="utf-8") as f:
|
||
conf = init_conf(yaml.load(f, Loader=yaml.FullLoader))
|
||
config = conf
|
||
if conf is None:
|
||
nonebot.logger.warning(
|
||
f"配置文件 {file_} 为空，已以默认配置创建，请在重启后更改为你所需的内容。"
|
||
)
|
||
conf = BasicConfig().dict()
|
||
return conf
|
||
|
||
|
||
def get_config(key: str, default=None):
|
||
"""获取配置项，优先级：bot > config > db > yaml"""
|
||
try:
|
||
bot = nonebot.get_bot()
|
||
except:
|
||
bot = None
|
||
|
||
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 load_from_yaml("config.yml"):
|
||
return load_from_yaml("config.yml")[key]
|
||
|
||
else:
|
||
return default
|
||
|
||
|
||
def init_conf(conf: dict) -> dict:
|
||
"""
|
||
初始化配置文件，确保配置文件中的必要字段存在，且不会冲突
|
||
Args:
|
||
conf:
|
||
|
||
Returns:
|
||
|
||
"""
|
||
# 若command_start中无""，则添加必要命令头，开启alconna_use_command_start防止冲突
|
||
# 以下内容由于issue #53 被注释
|
||
# if "" not in conf.get("command_start", []):
|
||
# conf["alconna_use_command_start"] = True
|
||
return conf
|
||
pass
|