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:
        if matcher.plugin_name not in plugins2settings_manager.keys():
            _plugin = matcher.plugin
            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,
                        )
        _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())} 个非限制插件.")