mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-acgnshow.git
synced 2024-09-20 21:22:35 +00:00
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
|
from nonebot import require, on_endswith
|
||
|
from nonebot.adapters import satori
|
||
|
from nonebot.adapters.onebot.v11 import MessageSegment
|
||
|
from nonebot.typing import T_State
|
||
|
from typing import Optional
|
||
|
from .acgnapis import *
|
||
|
from nonebot_plugin_htmlrender import text_to_pic, template_to_pic
|
||
|
from nonebot_plugin_alconna import on_alconna
|
||
|
from arclet.alconna import Alconna, Args
|
||
|
from .config import RES_PATH, TEMPLATE_NAME, config
|
||
|
from .util import *
|
||
|
|
||
|
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 showcmd.finish("未指定地区")
|
||
|
if not page: page = 1
|
||
|
if not date: date = ""
|
||
|
regions_dict = get_regions_dict()
|
||
|
regionid = regions_dict.get(region,None)
|
||
|
if regionid == None: await showcmd.finish("未找到此地区") ; return
|
||
|
#await showcmd.send("日期:"+ date)
|
||
|
shows = get_shows_data(regionid,page=page,pagesize=config.pagesize)
|
||
|
# print(shows)
|
||
|
template = {
|
||
|
"shows":process_shows_data_to_template(shows),
|
||
|
"bgimage": choose_random_bgimage()
|
||
|
}
|
||
|
pic = await template_to_pic(RES_PATH,TEMPLATE_NAME,template)
|
||
|
# print(pic)
|
||
|
# a = Image.open(io.BytesIO(pic))
|
||
|
# a.save("template2pic.png", format="PNG")
|
||
|
await showcmd.finish(MessageSegment.image(pic))
|