mirror of
https://github.com/LiteyukiStudio/LiteyukiBot.git
synced 2024-11-11 11:07:24 +08:00
72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
|
from urllib.parse import quote
|
||
|
|
|
||
|
|
import aiohttp
|
||
|
|
from nonebot import require
|
||
|
|
|
||
|
|
from liteyuki.utils.event import get_user_id
|
||
|
|
from liteyuki.utils.base.language import Language
|
||
|
|
from liteyuki.utils.base.ly_typing import T_MessageEvent
|
||
|
|
from liteyuki.utils.base.resource import get_path
|
||
|
|
from liteyuki.utils.message.html_tool import template2image
|
||
|
|
|
||
|
|
require("nonebot_plugin_alconna")
|
||
|
|
|
||
|
|
from nonebot_plugin_alconna import UniMessage, on_alconna, Alconna, Args, Subcommand, Arparma, Option
|
||
|
|
|
||
|
|
crt_cmd = on_alconna(
|
||
|
|
Alconna(
|
||
|
|
"crt",
|
||
|
|
Subcommand(
|
||
|
|
"route",
|
||
|
|
Args["start", str, "沙坪坝"]["end", str, "上新街"],
|
||
|
|
alias=("r",),
|
||
|
|
help_text="查询两地之间的地铁路线"
|
||
|
|
),
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@crt_cmd.assign("route")
|
||
|
|
async def _(result: Arparma, event: T_MessageEvent):
|
||
|
|
# 获取语言
|
||
|
|
ulang = Language(get_user_id(event))
|
||
|
|
|
||
|
|
# 获取参数
|
||
|
|
start = quote(quote(result.other_args.get("start")))
|
||
|
|
end = quote(quote(result.other_args.get("end")))
|
||
|
|
|
||
|
|
# 判断参数语言
|
||
|
|
query_lang_code = "Eng" if ulang.lang_code not in ["zh-CN", "zh-HK"] else ""
|
||
|
|
if start.isalpha() and end.isalpha():
|
||
|
|
query_lang_code = "Eng"
|
||
|
|
|
||
|
|
# 构造请求 URL
|
||
|
|
url = f"https://www.cqmetro.cn/Front/html/TakeLine!queryYs{query_lang_code}TakeLine.action?entity.startStaName={start}&entity.endStaName={end}"
|
||
|
|
|
||
|
|
# 请求数据
|
||
|
|
async with aiohttp.ClientSession() as session:
|
||
|
|
async with session.get(url) as resp:
|
||
|
|
result = await resp.json()
|
||
|
|
|
||
|
|
# 检查结果/无则终止
|
||
|
|
if not result.get("result"):
|
||
|
|
await crt_cmd.send(ulang.get("crt.no_result"))
|
||
|
|
return
|
||
|
|
|
||
|
|
# 模板传参定义
|
||
|
|
templates = {
|
||
|
|
"data": {
|
||
|
|
"result": result["result"],
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
# 生成图片
|
||
|
|
image = await template2image(
|
||
|
|
template=get_path("templates/crt_route.html"),
|
||
|
|
templates=templates,
|
||
|
|
debug=True
|
||
|
|
)
|
||
|
|
|
||
|
|
# 发送图片
|
||
|
|
await crt_cmd.send(UniMessage.image(raw=image))
|