summaryrefslogtreecommitdiff
path: root/ATRI/plugins/saucenao/__init__.py
diff options
context:
space:
mode:
authorLint Action <[email protected]>2021-07-31 08:17:46 +0000
committerLint Action <[email protected]>2021-07-31 08:17:46 +0000
commit36d26d1dc61c36b4601aaf75e148060c5bcb98a7 (patch)
tree3b466c4b0db4b9e160c5390a1d1c7ddb0322660f /ATRI/plugins/saucenao/__init__.py
parent336eb9d9e98b0bee952c27a50820dbdb350bcc03 (diff)
downloadATRI-36d26d1dc61c36b4601aaf75e148060c5bcb98a7.tar.gz
ATRI-36d26d1dc61c36b4601aaf75e148060c5bcb98a7.tar.bz2
ATRI-36d26d1dc61c36b4601aaf75e148060c5bcb98a7.zip
:rotating_light: 自动进行代码格式化
Diffstat (limited to 'ATRI/plugins/saucenao/__init__.py')
-rw-r--r--ATRI/plugins/saucenao/__init__.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/ATRI/plugins/saucenao/__init__.py b/ATRI/plugins/saucenao/__init__.py
index 468b001..b55f18c 100644
--- a/ATRI/plugins/saucenao/__init__.py
+++ b/ATRI/plugins/saucenao/__init__.py
@@ -16,6 +16,7 @@ _search_flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇�
saucenao = SaouceNao().on_command("以图搜图", "透过一张图搜索可能的来源")
+
@saucenao.args_parser # type: ignore
async def _get_img(bot: Bot, event: MessageEvent, state: T_State):
msg = str(event.message).strip()
@@ -27,16 +28,18 @@ async def _get_img(bot: Bot, event: MessageEvent, state: T_State):
else:
state["img"] = msg
+
@saucenao.handle()
async def _ready_search(bot: Bot, event: MessageEvent, state: T_State):
user_id = event.get_user_id()
if not _search_flmt.check(user_id):
await saucenao.finish(_search_flmt_notice)
-
+
msg = str(event.message).strip()
if msg:
state["img"] = msg
+
@saucenao.got("img", "图呢?")
async def _deal_search(bot: Bot, event: MessageEvent, state: T_State):
user_id = event.get_user_id()
@@ -44,7 +47,7 @@ async def _deal_search(bot: Bot, event: MessageEvent, state: T_State):
img = findall(r"url=(.*?)]", msg)
if not img:
await saucenao.reject("请发送图片而不是其他东西!!")
-
+
a = SaouceNao(SauceNAO.key)
result = f"> {MessageSegment.at(user_id)}" + await a.search(img[0])
_search_flmt.start_cd(user_id)