mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
122 lines
4.9 KiB
Python
122 lines
4.9 KiB
Python
|
from types import ModuleType
|
|||
|
|
from typing import Any
|
|||
|
|
|
|||
|
|
from services import logger
|
|||
|
|
from utils.manager import (
|
|||
|
|
plugin_data_manager,
|
|||
|
|
plugins2settings_manager,
|
|||
|
|
plugins2cd_manager,
|
|||
|
|
plugins2block_manager,
|
|||
|
|
plugins2count_manager,
|
|||
|
|
)
|
|||
|
|
from configs.config import Config
|
|||
|
|
from utils.manager.models import (
|
|||
|
|
PluginType,
|
|||
|
|
PluginSetting,
|
|||
|
|
PluginCd,
|
|||
|
|
PluginData,
|
|||
|
|
PluginBlock,
|
|||
|
|
PluginCount,
|
|||
|
|
)
|
|||
|
|
from utils.utils import get_matchers
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_attr(module: ModuleType, name: str, default: Any = None) -> Any:
|
|||
|
|
"""
|
|||
|
|
说明:
|
|||
|
|
获取属性
|
|||
|
|
参数:
|
|||
|
|
:param module: module
|
|||
|
|
:param name: name
|
|||
|
|
:param default: default
|
|||
|
|
"""
|
|||
|
|
if hasattr(module, name):
|
|||
|
|
return getattr(module, name)
|
|||
|
|
return default
|
|||
|
|
|
|||
|
|
|
|||
|
|
def init_plugin_info():
|
|||
|
|
|
|||
|
|
for matcher in [x for x in get_matchers(True) if x.plugin]:
|
|||
|
|
try:
|
|||
|
|
plugin = matcher.plugin
|
|||
|
|
metadata = plugin.metadata
|
|||
|
|
if hasattr(plugin, "module"):
|
|||
|
|
module = plugin.module
|
|||
|
|
plugin_model = matcher.plugin_name
|
|||
|
|
plugin_name = (
|
|||
|
|
metadata.name
|
|||
|
|
if metadata and metadata.name
|
|||
|
|
else get_attr(module, "__zx_plugin_name__", matcher.plugin_name)
|
|||
|
|
)
|
|||
|
|
if "[Admin]" in plugin_name:
|
|||
|
|
plugin_type = PluginType.ADMIN
|
|||
|
|
plugin_name = plugin_name.replace("[Admin]", "")
|
|||
|
|
elif "[Hidden]" in plugin_name:
|
|||
|
|
plugin_type = PluginType.HIDDEN
|
|||
|
|
plugin_name = plugin_name.replace("[Hidden]", "")
|
|||
|
|
elif "[Superuser]" in plugin_name:
|
|||
|
|
plugin_type = PluginType.SUPERUSER
|
|||
|
|
plugin_name = plugin_name.replace("[Superuser]", "")
|
|||
|
|
else:
|
|||
|
|
plugin_type = PluginType.NORMAL
|
|||
|
|
plugin_usage = (
|
|||
|
|
metadata.usage
|
|||
|
|
if metadata and metadata.usage
|
|||
|
|
else get_attr(module, "__plugin_usage__")
|
|||
|
|
)
|
|||
|
|
plugin_des = (
|
|||
|
|
metadata.description
|
|||
|
|
if metadata and metadata.description
|
|||
|
|
else get_attr(module, "__plugin_des__")
|
|||
|
|
)
|
|||
|
|
menu_type = get_attr(module, "__plugin_type__") or ("normal",)
|
|||
|
|
plugin_setting = get_attr(module, "__plugin_settings__")
|
|||
|
|
if plugin_setting:
|
|||
|
|
plugin_setting = PluginSetting(**plugin_setting)
|
|||
|
|
plugin_setting.plugin_type = menu_type
|
|||
|
|
plugin_task = get_attr(module, "__plugin_task__")
|
|||
|
|
plugin_version = get_attr(module, "__plugin_version__")
|
|||
|
|
plugin_author = get_attr(module, "__plugin_author__")
|
|||
|
|
plugin_cd = get_attr(module, "__plugin_cd_limit__")
|
|||
|
|
if plugin_cd:
|
|||
|
|
plugin_cd = PluginCd(**plugin_cd)
|
|||
|
|
plugin_block = get_attr(module, "__plugin_block_limit__")
|
|||
|
|
if plugin_block:
|
|||
|
|
plugin_block = PluginBlock(**plugin_block)
|
|||
|
|
plugin_count = get_attr(module, "__plugin_count_limit__")
|
|||
|
|
if plugin_count:
|
|||
|
|
plugin_count = PluginCount(**plugin_count)
|
|||
|
|
plugin_resources = get_attr(module, "__plugin_resources__")
|
|||
|
|
plugin_configs = get_attr(module, "__plugin_configs__")
|
|||
|
|
if settings := plugins2settings_manager.get(plugin_model):
|
|||
|
|
plugin_setting = settings
|
|||
|
|
if plugin_cd_limit := plugins2cd_manager.get(plugin_model):
|
|||
|
|
plugin_cd = plugin_cd_limit
|
|||
|
|
if plugin_block_limit := plugins2block_manager.get(plugin_model):
|
|||
|
|
plugin_block = plugin_block_limit
|
|||
|
|
if plugin_count_limit := plugins2count_manager.get(plugin_model):
|
|||
|
|
plugin_count = plugin_count_limit
|
|||
|
|
if plugin_cfg := Config.get(plugin_model):
|
|||
|
|
plugin_configs = plugin_cfg
|
|||
|
|
plugin_data = PluginData(
|
|||
|
|
model=plugin_model,
|
|||
|
|
name=plugin_name.strip(),
|
|||
|
|
plugin_type=plugin_type,
|
|||
|
|
usage=plugin_usage,
|
|||
|
|
des=plugin_des,
|
|||
|
|
task=plugin_task,
|
|||
|
|
menu_type=menu_type,
|
|||
|
|
version=plugin_version,
|
|||
|
|
author=plugin_author,
|
|||
|
|
plugin_setting=plugin_setting,
|
|||
|
|
plugin_cd=plugin_cd,
|
|||
|
|
plugin_block=plugin_block,
|
|||
|
|
plugin_count=plugin_count,
|
|||
|
|
plugin_resources=plugin_resources,
|
|||
|
|
plugin_configs=plugin_configs,
|
|||
|
|
)
|
|||
|
|
plugin_data_manager.add_plugin_info(plugin_data)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"构造插件数据失败 {matcher.plugin_name} - {type(e)}：{e}")
|