mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
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],
|
|
is_show=True,
|
|
)
|
|
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
|