zhenxun_bot/basic_plugins/init_plugin_config/init_plugins_data.py

from ruamel.yaml import YAML
from utils.manager import plugins_manager
from utils.utils import get_matchers
from services.log import logger
from configs.path_config import DATA_PATH

try:
    import ujson as json
except ModuleNotFoundError:
    import json


_yaml = YAML(typ="safe")


def init_plugins_data():
    """
    初始化插件数据信息
    """
    plugin2data_file = DATA_PATH / "manager" / "plugin_manager.json"
    plugin2data_file.parent.mkdir(parents=True, exist_ok=True)
    for matcher in get_matchers(True):
        _plugin = matcher.plugin
        if not _plugin:
            continue
        metadata = _plugin.metadata
        try:
            _module = _plugin.module
        except AttributeError:
            if matcher.plugin_name not in plugins_manager.keys():
                plugins_manager.add_plugin_data(
                    matcher.plugin_name, matcher.plugin_name, error=True
                )
            else:
                plugins_manager[matcher.plugin_name].error = True
        else:
            try:
                plugin_version = None
                if metadata:
                    plugin_version = metadata.extra.get("version")
                if not plugin_version and hasattr(_module, "__plugin_version__"):
                    plugin_version = _module.__getattribute__("__plugin_version__")
                if metadata:
                    plugin_name = metadata.name
                else:
                    try:
                        plugin_name = _module.__getattribute__("__zx_plugin_name__")
                    except AttributeError:
                        plugin_name = matcher.plugin_name
                plugin_author = None
                if metadata:
                    plugin_author = metadata.extra.get('author')
                if not plugin_author and hasattr(_module, "__plugin_author__"):
                    plugin_author = _module.__getattribute__("__plugin_author__")
                if matcher.plugin_name in plugins_manager.keys():
                    plugins_manager[matcher.plugin_name].error = False
                if matcher.plugin_name not in plugins_manager.keys():
                    plugins_manager.add_plugin_data(
                        matcher.plugin_name,
                        plugin_name=plugin_name,
                        author=plugin_author,
                        version=plugin_version,
                    )
                    # metadata不检测version
                elif isinstance(plugin_version, str) or plugins_manager[matcher.plugin_name].version is None or (
                    plugin_version is not None
                    and plugin_version > float(plugins_manager[matcher.plugin_name].version)
                ):
                    plugins_manager[matcher.plugin_name].plugin_name = plugin_name
                    plugins_manager[matcher.plugin_name].author = plugin_author
                    plugins_manager[matcher.plugin_name].version = plugin_version
                # if matcher.plugin_name in _data.keys():
                #     plugins_manager[matcher.plugin_name].error = _data[matcher.plugin_name]["error"]
                #     plugins_manager.set_module_data(
                #         matcher.plugin_name, "error", _data[matcher.plugin_name]["error"]
                #     )
                #     plugins_manager.set_module_data(
                #         matcher.plugin_name, "plugin_name", _data[matcher.plugin_name]["plugin_name"]
                #     )
            except Exception as e:
                logger.error(f"插件数据 {matcher.plugin_name} 加载发生错误 {type(e)}：{e}")
    plugins_manager.save()