zhenxun_bot/zhenxun/builtin_plugins/help/_utils.py

from collections.abc import Callable

from zhenxun.utils.enum import PluginType
from zhenxun.models.plugin_info import PluginInfo
from zhenxun.models.group_console import GroupConsole


async def sort_type() -> dict[str, list[PluginInfo]]:
    """
    对插件按照菜单类型分类
    """
    data = await PluginInfo.filter(
        menu_type__not="",
        load_status=True,
        plugin_type__in=[PluginType.NORMAL, PluginType.DEPENDANT],
    )
    sort_data = {}
    for plugin in data:
        menu_type = plugin.menu_type or "normal"
        if menu_type == "normal":
            menu_type = "功能"
        if not sort_data.get(menu_type):
            sort_data[menu_type] = []
        sort_data[menu_type].append(plugin)
    return sort_data


async def classify_plugin(group_id: str | None, handle: Callable) -> dict[str, list]:
    """对插件进行分类并判断状态

    参数:
        group_id: 群组id

    返回:
        dict[str, list[Item]]: 分类插件数据
    """
    sort_data = await sort_type()
    classify: dict[str, list] = {}
    group = await GroupConsole.get_or_none(group_id=group_id) if group_id else None
    for menu, value in sort_data.items():
        for plugin in value:
            if not classify.get(menu):
                classify[menu] = []
            classify[menu].append(handle(plugin, group))
    return classify