mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-acgnshow.git
synced 2024-11-11 01:27:39 +08:00
53 lines
1.8 KiB
Python
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() |