mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
|
from utils.image_utils import CreateImg
|
|||
|
|
from configs.path_config import IMAGE_PATH
|
|||
|
|
from services.log import logger
|
|||
|
|
from utils.utils import get_matchers
|
|||
|
|
from nonebot.adapters.cqhttp import Bot
|
|||
|
|
from pathlib import Path
|
|||
|
|
from nonebot import Driver
|
|||
|
|
import asyncio
|
|||
|
|
import nonebot
|
|||
|
|
|
|||
|
|
|
|||
|
|
driver: Driver = nonebot.get_driver()
|
|||
|
|
|
|||
|
|
background = Path(IMAGE_PATH) / "background" / "0.png"
|
|||
|
|
|
|||
|
|
superuser_help_image = Path(IMAGE_PATH) / "superuser_help.png"
|
|||
|
|
|
|||
|
|
|
|||
|
|
@driver.on_bot_connect
|
|||
|
|
async def create_help_image(bot: Bot = None):
|
|||
|
|
"""
|
|||
|
|
创建超级用户帮助图片
|
|||
|
|
"""
|
|||
|
|
await asyncio.get_event_loop().run_in_executor(None, _create_help_image)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _create_help_image():
|
|||
|
|
"""
|
|||
|
|
创建管理员帮助图片
|
|||
|
|
"""
|
|||
|
|
_matchers = get_matchers()
|
|||
|
|
_plugin_name_list = []
|
|||
|
|
width = 0
|
|||
|
|
help_str = "超级用户帮助\n\n* 注: ‘*’ 代表可有多个相同参数 ‘?’ 代表可省略该参数 *\n\n"
|
|||
|
|
tmp_img = CreateImg(0, 0, plain_text='1', font_size=24)
|
|||
|
|
plugin_name = ""
|
|||
|
|
for matcher in _matchers:
|
|||
|
|
try:
|
|||
|
|
_plugin = nonebot.plugin.get_plugin(matcher.module)
|
|||
|
|
_module = _plugin.module
|
|||
|
|
try:
|
|||
|
|
plugin_name = _module.__getattribute__("__zx_plugin_name__")
|
|||
|
|
except AttributeError:
|
|||
|
|
continue
|
|||
|
|
is_superuser_usage = False
|
|||
|
|
try:
|
|||
|
|
_ = _module.__getattribute__("__plugin_superuser_usage__")
|
|||
|
|
is_superuser_usage = True
|
|||
|
|
except AttributeError:
|
|||
|
|
pass
|
|||
|
|
if (
|
|||
|
|
("[superuser]" in plugin_name.lower() or is_superuser_usage)
|
|||
|
|
and plugin_name != "超级用户帮助 [Superuser]"
|
|||
|
|
and plugin_name not in _plugin_name_list
|
|||
|
|
and "[hidden]" not in plugin_name.lower()
|
|||
|
|
):
|
|||
|
|
_plugin_name_list.append(plugin_name)
|
|||
|
|
try:
|
|||
|
|
plugin_des = _module.__getattribute__("__plugin_des__")
|
|||
|
|
except AttributeError:
|
|||
|
|
plugin_des = '_'
|
|||
|
|
plugin_cmd = _module.__getattribute__("__plugin_cmd__")
|
|||
|
|
if is_superuser_usage:
|
|||
|
|
plugin_cmd = [x for x in plugin_cmd if "[_superuser]" in x]
|
|||
|
|
plugin_cmd = " / ".join(plugin_cmd).replace('[_superuser]', '').strip()
|
|||
|
|
help_str += f"{plugin_des} -> {plugin_cmd}\n\n"
|
|||
|
|
x = tmp_img.getsize(f"{plugin_des} -> {plugin_cmd}")[0]
|
|||
|
|
width = width if width > x else x
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(
|
|||
|
|
f"获取超级用户插件 {matcher.module}: {plugin_name} 设置失败... {type(e)}：{e}"
|
|||
|
|
)
|
|||
|
|
height = len(help_str.split("\n")) * 33
|
|||
|
|
width += 500
|
|||
|
|
A = CreateImg(width, height, font_size=24)
|
|||
|
|
_background = CreateImg(width, height, background=background)
|
|||
|
|
A.text((300, 140), help_str)
|
|||
|
|
A.paste(_background, alpha=True)
|
|||
|
|
A.save(superuser_help_image)
|
|||
|
|
logger.info(f"已成功加载 {len(_plugin_name_list)} 条超级用户命令")
|