zhenxun_bot/plugins/help/__init__.py
2021-06-30 19:50:55 +08:00

46 lines
1.5 KiB
Python

from nonebot import on_command
from nonebot.adapters.cqhttp import Bot, GroupMessageEvent, PrivateMessageEvent
from nonebot.typing import T_State
from nonebot.rule import to_me
from configs.path_config import DATA_PATH
from utils.init_result import image
import os
from .data_source import create_help_img, create_group_help_img, get_plugin_help
from nonebot import require
from utils.utils import get_message_text
export = require("nonebot_plugin_manager")
__plugin_name__ = '帮助'
if not os.path.exists(DATA_PATH + 'group_help/'):
os.mkdir(DATA_PATH + 'group_help/')
create_help_img()
for file in os.listdir(DATA_PATH + 'group_help/'):
os.remove(DATA_PATH + 'group_help/' + file)
_help = on_command("功能", rule=to_me(), aliases={'help', '帮助'}, priority=1, block=True)
@_help.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
msg = get_message_text(event.json())
if not msg:
if not os.path.exists(DATA_PATH + f'group_help/{event.group_id}.png'):
create_group_help_img(event.group_id)
await _help.finish(image(abspath=DATA_PATH + f'group_help/{event.group_id}.png'))
else:
await _help.finish(get_plugin_help(msg))
@_help.handle()
async def _(bot: Bot, event: PrivateMessageEvent, state: T_State):
msg = get_message_text(event.json())
if not msg:
await _help.finish(image('help.png'))
else:
await _help.finish(get_plugin_help(msg))