diff options
author | Lint Action <[email protected]> | 2021-08-26 02:45:06 +0000 |
---|---|---|
committer | Lint Action <[email protected]> | 2021-08-26 02:45:06 +0000 |
commit | b91bb2c0ccce4443c1cfe6b7de5ba33439ca6cf2 (patch) | |
tree | acda55ba0f417b0d992998cd254464cfb00777f7 /ATRI/plugins/saucenao/__init__.py | |
parent | b4667a5c838adb7a4d0a53eeee992e4784231895 (diff) | |
download | ATRI-b91bb2c0ccce4443c1cfe6b7de5ba33439ca6cf2.tar.gz ATRI-b91bb2c0ccce4443c1cfe6b7de5ba33439ca6cf2.tar.bz2 ATRI-b91bb2c0ccce4443c1cfe6b7de5ba33439ca6cf2.zip |
:rotating_light: 自动进行代码格式化
Diffstat (limited to 'ATRI/plugins/saucenao/__init__.py')
-rw-r--r-- | ATRI/plugins/saucenao/__init__.py | 7 |
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) |