zhenxun_bot/basic_plugins/init_plugin_config/init_plugins_settings.py

from pathlib import Path
from ruamel.yaml import round_trip_load, round_trip_dump, YAML
from utils.manager import plugins2settings_manager, admin_manager
from services.log import logger
from utils.utils import get_matchers
from ruamel import yaml
import nonebot


_yaml = YAML(typ="safe")


def init_plugins_settings(data_path: str):
    """
    初始化插件设置，从插件中获取 __zx_plugin_name__，__plugin_cmd__，__plugin_settings__
    """
    plugins2settings_file = data_path / "configs" / "plugins2settings.yaml"
    plugins2settings_file.parent.mkdir(exist_ok=True, parents=True)
    _matchers = get_matchers(True)
    _tmp_module = {}
    _tmp = []
    for x in plugins2settings_manager.keys():
        try:
            _plugin = nonebot.plugin.get_plugin(x)
            _module = _plugin.module
            metadata = _plugin.metadata
            plugin_name = (
                metadata.name
                if metadata
                else _module.__getattribute__("__zx_plugin_name__")
            )
            _tmp_module[x] = plugin_name
        except (KeyError, AttributeError) as e:
            logger.warning(f"配置文件 模块：{x} 获取 plugin_name 失败...{e}")
            _tmp_module[x] = ""
    for matcher in _matchers:
        try:
            if matcher.plugin_name not in plugins2settings_manager.keys():
                _plugin = matcher.plugin
                if not _plugin:
                    continue
                metadata = _plugin.metadata
                try:
                    _module = _plugin.module
                except AttributeError:
                    logger.warning(f"插件 {matcher.plugin_name} 加载失败...，插件控制未加载.")
                else:
                    try:
                        if metadata:
                            plugin_name = metadata.name
                        else:
                            plugin_name = _module.__getattribute__("__zx_plugin_name__")
                        if "[admin]" in plugin_name.lower():
                            try:
                                admin_settings = _module.__getattribute__(
                                    "__plugin_settings__"
                                )
                                level = admin_settings["admin_level"]
                                cmd = admin_settings.get("cmd")
                            except (AttributeError, KeyError):
                                level = 5
                                cmd = None
                            if level is None:
                                level = 5
                            admin_manager.add_admin_plugin_settings(
                                matcher.plugin_name, cmd, level
                            )
                        if (
                            "[hidden]" in plugin_name.lower()
                            or "[admin]" in plugin_name.lower()
                            or "[superuser]" in plugin_name.lower()
                            or matcher.plugin_name in plugins2settings_manager.keys()
                        ):
                            continue
                    except AttributeError:
                        if matcher.plugin_name not in _tmp:
                            logger.warning(
                                f"获取插件 {matcher.plugin_name} __zx_plugin_name__ 失败...，插件控制未加载."
                            )
                    else:
                        _tmp_module[matcher.plugin_name] = plugin_name
                        try:
                            plugin_settings = _module.__getattribute__(
                                "__plugin_settings__"
                            )
                        except AttributeError:
                            plugin_settings = {"cmd": [matcher.plugin_name, plugin_name]}
                        if not plugin_settings.get("cost_gold"):
                            plugin_settings["cost_gold"] = 0
                        if (
                            plugin_settings.get("cmd") is not None
                            and plugin_name not in plugin_settings["cmd"]
                        ):
                            plugin_settings["cmd"].append(plugin_name)
                        if plugins2settings_manager.get(
                            matcher.plugin_name
                        ) and plugins2settings_manager[matcher.plugin_name].get(
                            "plugin_type"
                        ):
                            plugin_type = tuple(
                                plugins2settings_manager.get_plugin_data(
                                    matcher.plugin_name
                                )["plugin_type"]
                            )
                        else:
                            try:
                                plugin_type = _module.__getattribute__("__plugin_type__")
                            except AttributeError:
                                plugin_type = ("normal",)
                        if plugin_settings and matcher.plugin_name:
                            plugins2settings_manager.add_plugin_settings(
                                matcher.plugin_name,
                                plugin_type=plugin_type,
                                **plugin_settings,
                            )
        except Exception as e:
            logger.error(f'{matcher.plugin_name} 初始化 plugin_settings 发生错误 {type(e)}：{e}')
        _tmp.append(matcher.plugin_name)
    _tmp_data = {"PluginSettings": plugins2settings_manager.get_data()}
    with open(plugins2settings_file, "w", encoding="utf8") as wf:
        yaml.dump(_tmp_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
    _data = round_trip_load(open(plugins2settings_file, encoding="utf8"))
    _data["PluginSettings"].yaml_set_start_comment(
        """# 模块与对应命令和对应群权限
# 用于生成帮助图片 和 开关功能
# key：模块名称
# level：需要的群等级
# default_status：加入群时功能的默认开关状态
# limit_superuser: 功能状态是否限制超级用户
# cmd: 关闭[cmd] 都会触发命令 关闭对应功能，cmd列表第一个词为统计的功能名称
# plugin_type: 帮助类别 示例：('原神相关',) 或 ('原神相关', 1)，1代表帮助命令列向排列，否则为横向排列""",
        indent=2,
    )
    for plugin in _data["PluginSettings"].keys():
        _data["PluginSettings"][plugin].yaml_set_start_comment(
            f"{plugin}：{_tmp_module[plugin]}", indent=2
        )
    with open(plugins2settings_file, "w", encoding="utf8") as wf:
        round_trip_dump(_data, wf, Dumper=yaml.RoundTripDumper, allow_unicode=True)
    logger.info(f"已成功加载 {len(plugins2settings_manager.get_data())} 个非限制插件.")