Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 06:12:53 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

fix what_anime

Browse Source
This commit is contained in:
HibiKier 2022-05-21 19:33:39 +08:00
parent 324b58c21b
commit 33d1334174
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
plugins/what_anime/__init__.py
View File

@ -1,5 +1,6 @@
from nonebot.adapters.onebot.v11 import MessageEvent, GroupMessageEvent, Message, Bot from nonebot.adapters.onebot.v11 import MessageEvent, GroupMessageEvent, Message, Bot
from nonebot.internal.params import ArgStr from nonebot.internal.params import ArgStr, Arg
from nonebot.params import CommandArg
from .data_source import get_anime from .data_source import get_anime
from nonebot import on_command from nonebot import on_command
@ -32,16 +33,18 @@ what_anime = on_command("识番", priority=5, block=True)
@what_anime.handle() @what_anime.handle()
async def _(bot: Bot, event: MessageEvent, state: T_State): async def _(bot: Bot, event: MessageEvent, state: T_State, args: Message = CommandArg()):
img_url = get_message_img(event.json()) img_url = get_message_img(event.json())
if img_url: if img_url:
state["img_url"] = img_url[0] state["img_url"] = args
@what_anime.got("img_url", prompt="虚空识番？来图来图GKD") @what_anime.got("img_url", prompt="虚空识番？来图来图GKD")
async def _(bot: Bot, event: MessageEvent, state: T_State, img_url: str = ArgStr("img_url")): async def _(bot: Bot, event: MessageEvent, state: T_State, img_url: Message = Arg("img_url")):
img_url = get_message_img(img_url)
if not img_url: if not img_url:
await what_anime.reject_arg("img_url", "发送的必须是图片！") await what_anime.reject_arg("img_url", "发送的必须是图片！")
img_url = img_url[0]
await what_anime.send("开始识别.....") await what_anime.send("开始识别.....")
anime_data_report = await get_anime(img_url) anime_data_report = await get_anime(img_url)
if anime_data_report: if anime_data_report:

1
pyproject.toml
View File

@ -14,7 +14,6 @@ url = "https://mirrors.aliyun.com/pypi/simple/"
python = "^3.8" python = "^3.8"
nonebot2 = "^2.0.0-beta.2" nonebot2 = "^2.0.0-beta.2"
nonebot-adapter-onebot = "^2.0.0-beta.1" nonebot-adapter-onebot = "^2.0.0-beta.1"
bilibili-api = "^9.1.0"
aiofiles = "^0.8.0" aiofiles = "^0.8.0"
aiohttp = "3.7.4.post0" aiohttp = "3.7.4.post0"
beautifulsoup4 = "4.9.3" beautifulsoup4 = "4.9.3"