mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 11:29:39 +08:00
99 lines
4.4 KiB
Python
99 lines
4.4 KiB
Python
from ..aamgr import AbstractCommandNode, Context
|
||
import json
|
||
|
||
|
||
def config_operation(cmd, params):
|
||
reply = []
|
||
import pkg.utils.context
|
||
config = pkg.utils.context.get_config()
|
||
reply_str = ""
|
||
if len(params) == 0:
|
||
reply = ["[bot]err:请输入!cmd cfg查看使用方法"]
|
||
else:
|
||
cfg_name = params[0]
|
||
if cfg_name == 'all':
|
||
reply_str = "[bot]所有配置项:\n\n"
|
||
for cfg in dir(config):
|
||
if not cfg.startswith('__') and not cfg == 'logging':
|
||
# 根据配置项类型进行格式化，如果是字典则转换为json并格式化
|
||
if isinstance(getattr(config, cfg), str):
|
||
reply_str += "{}: \"{}\"\n".format(cfg, getattr(config, cfg))
|
||
elif isinstance(getattr(config, cfg), dict):
|
||
# 不进行unicode转义，并格式化
|
||
reply_str += "{}: {}\n".format(cfg,
|
||
json.dumps(getattr(config, cfg),
|
||
ensure_ascii=False, indent=4))
|
||
else:
|
||
reply_str += "{}: {}\n".format(cfg, getattr(config, cfg))
|
||
reply = [reply_str]
|
||
else:
|
||
cfg_entry_path = cfg_name.split('.')
|
||
|
||
try:
|
||
if len(params) == 1:
|
||
cfg_entry = getattr(config, cfg_entry_path[0])
|
||
if len(cfg_entry_path) > 1:
|
||
for i in range(1, len(cfg_entry_path)):
|
||
cfg_entry = cfg_entry[cfg_entry_path[i]]
|
||
|
||
if isinstance(cfg_entry, str):
|
||
reply_str = "[bot]配置项{}: \"{}\"\n".format(cfg_name, cfg_entry)
|
||
elif isinstance(cfg_entry, dict):
|
||
reply_str = "[bot]配置项{}: {}\n".format(cfg_name,
|
||
json.dumps(cfg_entry,
|
||
ensure_ascii=False, indent=4))
|
||
else:
|
||
reply_str = "[bot]配置项{}: {}\n".format(cfg_name, cfg_entry)
|
||
reply = [reply_str]
|
||
else:
|
||
cfg_value = " ".join(params[1:])
|
||
# 类型转换，如果是json则转换为字典
|
||
if cfg_value == 'true':
|
||
cfg_value = True
|
||
elif cfg_value == 'false':
|
||
cfg_value = False
|
||
elif cfg_value.isdigit():
|
||
cfg_value = int(cfg_value)
|
||
elif cfg_value.startswith('{') and cfg_value.endswith('}'):
|
||
cfg_value = json.loads(cfg_value)
|
||
else:
|
||
try:
|
||
cfg_value = float(cfg_value)
|
||
except ValueError:
|
||
pass
|
||
|
||
cfg_entry = getattr(config, cfg_entry_path[0])
|
||
if len(cfg_entry_path) > 1:
|
||
for i in range(1, len(cfg_entry_path) - 1):
|
||
cfg_entry = cfg_entry[cfg_entry_path[i]]
|
||
if isinstance(cfg_entry[cfg_entry_path[-1]], type(cfg_value)):
|
||
cfg_entry[cfg_entry_path[-1]] = cfg_value
|
||
reply = ["[bot]配置项{}修改成功".format(cfg_name)]
|
||
else:
|
||
reply = ["[bot]err:配置项{}类型不匹配".format(cfg_name)]
|
||
else:
|
||
setattr(config, cfg_entry_path[0], cfg_value)
|
||
reply = ["[bot]配置项{}修改成功".format(cfg_name)]
|
||
except AttributeError:
|
||
reply = ["[bot]err:未找到配置项 {}".format(cfg_name)]
|
||
except ValueError:
|
||
reply = ["[bot]err:未找到配置项 {}".format(cfg_name)]
|
||
# else:
|
||
# reply = ["[bot]err:未找到配置项 {}".format(cfg_name)]
|
||
|
||
return reply
|
||
|
||
|
||
@AbstractCommandNode.register(
|
||
parent=None,
|
||
name="cfg",
|
||
description="配置项管理",
|
||
usage="!cfg <配置项> [配置值]\n!cfg all",
|
||
aliases=[],
|
||
privilege=2
|
||
)
|
||
class CfgCommand(AbstractCommandNode):
|
||
@classmethod
|
||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
||
return True, config_operation(ctx.command, ctx.params)
|
||
|