zhenxun_bot/plugins/help/data_source.py

from util.img_utils import CreateImg
from configs.path_config import IMAGE_PATH, DATA_PATH
import ujson as json
import os
from .config import *
from nonebot import require
from configs.config import INITIAL_OPEN_CASE_COUNT, INITIAL_SETU_PROBABILITY, ADMIN_DEFAULT_AUTH

export = require("nonebot_plugin_manager")

width = 1500
e_height = 0
u_height = 850
o_height = 1500
# f_height =


def create_help_img():
    if os.path.exists(IMAGE_PATH + 'help.png'):
        os.remove(IMAGE_PATH + 'help.png')
    h = (100 + len(utility_help) * 24 + len(entertainment_help) * 24 + len(other_help) * 24) * 2
    A = CreateImg(width, h - 200, font_size=24)
    e = CreateImg(width, len(entertainment_help) * 42, font_size=24)
    rst = ''
    i = 0
    for cmd in entertainment_help:
        rst += f'{i + 1}.{entertainment_help[cmd]}\n'
        i += 1
    e.text((10, 10), '娱乐功能：')
    e.text((40, 40), rst)
    u = CreateImg(width, len(utility_help) * 40 + 50, font_size=24, color='black')
    rst = ''
    i = 0
    for cmd in utility_help:
        rst += f'{i + 1}.{utility_help[cmd]}\n'
        i += 1
    u.text((10, 10), '实用功能：', fill=(255, 255, 255))
    u.text((40, 40), rst, fill=(255, 255, 255))
    o = CreateImg(width, len(other_help) * 40, font_size=24)
    rst = ''
    i = 0
    for i in range(len(other_help)):
        rst += f'{i + 1}.{other_help[i]}\n'
        i += 1
    o.text((10, 10), '其他功能：')
    o.text((40, 40), rst)
    A.paste(e, (0, 0))
    A.paste(u, (0, u_height))
    A.paste(o, (0, o_height))
    A.text((10, h * 0.76), '大部分交互功能可以通过输入‘取消’，‘算了’来取消当前交互\n对我说 “指令名 帮助” 获取对应详细帮助\n'
                           '可以通过 “滴滴滴- 后接内容” 联系管理员（有趣的想法尽管来吧！<还有Bug和建议>）\n[群管理员请看 管理员帮助（群主与管理员自带 5 级权限）]')
    A.text((10, h * 0.81), f"【注】「色图概率：好感度 + 70%\n"
                           f"\t\t每 3 点好感度 + 1次开箱，初始 20 次\n"
                           f"\t\t开启/关闭功能只需输入‘开启/关闭 指令名’（每个功能的第一个指令）」\n"
                           f"\t\t示例：开启签到")
    A.save(IMAGE_PATH + 'help.png')


def create_group_help_img(group_id: int):
    group_id = str(group_id)
    try:
        with open(DATA_PATH + 'manager/plugin_list.json', 'r', encoding='utf8') as f:
            plugin_list = json.load(f)
    except (ValueError, FileNotFoundError):
        pass
    h = (100 + len(utility_help) * 24 + len(entertainment_help) * 24 + len(other_help) * 24) * 2
    A = CreateImg(1200, h - 200, font_size=24)
    u = CreateImg(1200, len(utility_help) * 40, font_size=24, color='black')
    o = CreateImg(1200, len(other_help) * 40, font_size=24)
    e = CreateImg(width, len(entertainment_help) * 42, font_size=24)
    rst = ''
    i = 1
    for cmd in entertainment_help.keys():
        flag, dfg = parse_cmd(cmd, group_id, plugin_list)
        if dfg:
            cmd = rcmd(dfg)
        rst += f'【{flag}】{i}.{entertainment_help[cmd]}\n'
        i += 1
    e.text((10, 10), '娱乐功能：')
    e.text((40, 40), rst)

    rst = ''
    i = 1
    for cmd in utility_help.keys():
        flag, dfg = parse_cmd(cmd, group_id, plugin_list)
        rst += f'【{flag}】{i}.{utility_help[cmd]}\n'
        i += 1
    u.text((10, 10), '实用功能：', fill=(255, 255, 255))
    u.text((40, 40), rst, fill=(255, 255, 255))

    rst = ''
    for i in range(len(other_help)):
        rst += f'{i + 1}.{other_help[i]}\n'
    o.text((10, 10), '其他功能：')
    o.text((40, 40), rst)

    A.paste(e, (0, 0))
    A.paste(u, (0, u_height))
    A.paste(o, (0, o_height))
    # A.text((width, 10), f'总开关【{"√" if data["总开关"] else "×"}】')
    A.text((10, h * 0.76), '大部分交互功能可以通过输入‘取消’，‘算了’来取消当前交互\n对我说 “指令名 帮助” 获取对应详细帮助\n'
                           '可以通过 “滴滴滴- 后接内容” 联系管理员（有趣的想法尽管来吧！<还有Bug和建议>）'
                           f'\n[群管理员请看 管理员帮助（群主与管理员自带 {ADMIN_DEFAULT_AUTH} 级权限）]')
    A.text((10, h * 0.81), f"【注】「色图概率：好感度 + {int(INITIAL_SETU_PROBABILITY*100)}%\n"
                           f"\t\t每 3 点好感度 + 1次开箱，初始 {INITIAL_OPEN_CASE_COUNT} 次\n"
                           f"\t\t开启/关闭功能只需输入‘开启/关闭 指令名’（每个功能的第一个指令）」\n"
                           f"\t\t示例：开启签到\n"
                           f"\t\t可以通过管理员开关自动发送消息(早晚安等)\n"
                           f"\t\t^请查看管理员帮助^")
    A.save(DATA_PATH + f'group_help/{group_id}.png')


def parse_cmd(cmd, group_id, plugin_list):
    flag = '√'
    dfg = None
    if cmd.find('draw_card') != -1:
        lst = cmd.split('_')
        cmd = lst[0] + '_' + lst[1]
        dfg = lst[-1]
    elif cmd == 'pixiv_r':
        cmd = 'pixiv'
        dfg = 'r'
    elif cmd == 'pixiv_s':
        cmd = 'pixiv'
        dfg = 's'
    if group_id in plugin_list[cmd]:
        if not plugin_list[cmd][group_id]:
            flag = '×'
    if cmd in ['bt', 'reimu', 'nickname']:
        flag = '- '
    return flag, dfg


def rcmd(dfg):
    if dfg in ['prts', 'genshin', 'pretty', 'guardian', 'pcr', 'azur', 'onmyoji', 'fgo']:
        return 'draw_card_' + dfg
    if dfg == 'r':
        return 'pixiv_r'
    if dfg == 's':
        return 'pixiv_s'