1 files changed, 25 insertions, 27 deletions
diff --git a/ATRI/plugins/saucenao/__init__.py b/ATRI/plugins/saucenao/__init__.py
index f158cf9..b4b1497 100644
--- a/ATRI/plugins/saucenao/__init__.py
+++ b/ATRI/plugins/saucenao/__init__.py
@@ -21,55 +21,53 @@ __doc__ = """
   以图搜图 (pic)
 """
 
-saucenao = sv.on_command(
-  cmd='以图搜图',
-  docs=__doc__,
-  rule=is_in_service('以图搜图')
-)
+saucenao = sv.on_command(cmd="以图搜图", docs=__doc__, rule=is_in_service("以图搜图"))
+
 
 @saucenao.args_parser  # type: ignore
-async def _load_saucenao(bot: Bot, event: MessageEvent,
-                         state: T_State) -> None:
+async def _load_saucenao(bot: Bot, event: MessageEvent, state: T_State) -> None:
     msg = str(event.message)
-    quit_list = ['算了', '罢了', '不搜了']
+    quit_list = ["算了", "罢了", "不搜了"]
     if msg in quit_list:
-        await saucenao.finish('好吧...')
-    
+        await saucenao.finish("好吧...")
+
     if not msg:
-        await saucenao.reject('图呢？')
+        await saucenao.reject("图呢？")
     else:
-        state['pic_sau'] = msg
+        state["pic_sau"] = msg
+
 
 @saucenao.handle()
 async def _sauce_nao(bot: Bot, event: MessageEvent, state: T_State) -> None:
     msg = str(event.message).strip()
     if msg:
-        state['pic_sau'] = msg
+        state["pic_sau"] = msg
 
-@saucenao.got('pic_sau', prompt='图呢？')
+
+@saucenao.got("pic_sau", prompt="图呢？")
 async def _deal_saucenao(bot: Bot, event: MessageEvent, state: T_State) -> None:
-    msg = state['pic_sau']
-    img = re.findall(r'url=(.*?)]', msg)
+    msg = state["pic_sau"]
+    img = re.findall(r"url=(.*?)]", msg)
     if not img:
-        await saucenao.finish('请发送图片而不是其他东西！！')
-    
+        await saucenao.finish("请发送图片而不是其他东西！！")
+
     try:
         task = SauceNao(api_key=Config.SauceNAO.key)
         data = json.loads(await task.search(img[0]))
     except RequestTimeOut:
-        raise RequestTimeOut('Request failed!')
-    
-    res = data['results']
+        raise RequestTimeOut("Request failed!")
+
+    res = data["results"]
     result = list()
     for i in range(0, 3):
         data = res[i]
-        
+
         _result = dict()
-        _result['similarity'] = data['header']['similarity']
-        _result['index_name'] = data['header']['index_name']
-        _result['url'] = choice(data['data'].get('ext_urls', ['None']))
+        _result["similarity"] = data["header"]["similarity"]
+        _result["index_name"] = data["header"]["index_name"]
+        _result["url"] = choice(data["data"].get("ext_urls", ["None"]))
         result.append(_result)
-    
+
     msg0 = f"> {MessageSegment.at(event.user_id)}"
     for i in result:
         msg0 = msg0 + (
@@ -78,5 +76,5 @@ async def _deal_saucenao(bot: Bot, event: MessageEvent, state: T_State) -> None:
             f"Name: {i['index_name']}\n"
             f"URL: {i['url'].replace('https://', '')}"
         )
-    
+
     await saucenao.finish(Message(msg0))