nonebot-plugin-acgnshow/nonebot_plugin_acgnshow/acgnshower.py

53 lines
1.8 KiB
Python

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(
state: T_State, 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 = get_regions_dict()
regionid = regions_dict.get(region,None)
if regionid == None: await UniMessage("不支持此地区").send() ; return
#await showcmd.send("日期:"+ date)
shows = get_shows_data(regionid,page=page,pagesize=config.acgnshow_pagesize)
# print(shows)
try:
showsdata = process_shows_data_to_template(shows)
template = {
"shows": showsdata[0],
"bgimage": choose_random_bgimage(),
"global_data": showsdata[1]
}
pic = await template_to_pic(RES_PATH,TEMPLATE_NAME,template)
except:
await UniMessage("发生错误").send()
return
# print(pic)
# a = Image.open(io.BytesIO(pic))
# a.save("template2pic.png", format="PNG")
await UniMessage.image(pic).send()