nonebot-plugin-acgnshow/nonebot_plugin_acgnshow/acgnshower.py

import traceback

from nonebot.typing import T_State
from typing import Optional
from .acgnapis import *
from nonebot_plugin_htmlrender import template_to_pic
from nonebot_plugin_alconna import on_alconna
from nonebot_plugin_alconna.uniseg import UniMessage
from arclet.alconna import Alconna, Args
from .config import RES_PATH, TEMPLATE_NAME, config
from .util import *
from .__init__ import __plugin_meta__

showcmd = on_alconna(
    Alconna(
        "展览",
        Args["region?", str]["page?", int]["date?", str],
    )
)
showcmd.shortcut(
    r"(?P<region>.+?)展览\s*(?P<page>\d+)?\s*(?P<date>.+)?",
    {
            "prefix" : True,
            "command": "展览",
            "args"   : ["{region}", "{page}", "{date}"],
    }
)


@showcmd.handle()
async def find_show(
        region: Optional[str] = None, page: Optional[int] = None, date: Optional[str] = None,
):
    if not region: await UniMessage(__plugin_meta__.usage).send(); return
    if not page: page = 1
    if not date: date = ""
    regions_dict = await get_regions_dict()
    regionid = regions_dict.get(region, None)
    if regionid == None: await UniMessage("不支持此地区").send(); return
    # await showcmd.send("日期："+ date)
    shows = await get_shows_data(regionid, page=page, pagesize=config.acgnshow_pagesize)
    # print(shows)
    try:
        shows_data = process_shows_data_to_template(shows)
        template = {
                "shows"      : shows_data[0],
                "bgimage"    : choose_random_bgimage(),
                "global_data": shows_data[1]
        }
        pic = await template_to_pic(RES_PATH, TEMPLATE_NAME, template)
    except Exception as e:
        await UniMessage("发生错误").send()
        traceback.print_exc()
        return

    await UniMessage.image(raw=pic).send()