diff options
Diffstat (limited to 'ATRI/plugins/saucenao')
| -rw-r--r-- | ATRI/plugins/saucenao/__init__.py | 6 | ||||
| -rw-r--r-- | ATRI/plugins/saucenao/data_source.py | 2 |
2 files changed, 6 insertions, 2 deletions
diff --git a/ATRI/plugins/saucenao/__init__.py b/ATRI/plugins/saucenao/__init__.py index b55f18c..d0bf606 100644 --- a/ATRI/plugins/saucenao/__init__.py +++ b/ATRI/plugins/saucenao/__init__.py @@ -48,7 +48,11 @@ async def _deal_search(bot: Bot, event: MessageEvent, state: T_State): if not img: await saucenao.reject("请发送图片而不是其他东西！！") - a = SaouceNao(SauceNAO.key) + try: + a = SaouceNao(SauceNAO.key) + except Exception: + await saucenao.finish("失败了...") + result = f"> {MessageSegment.at(user_id)}" + await a.search(img[0]) _search_flmt.start_cd(user_id) await saucenao.finish(Message(result)) diff --git a/ATRI/plugins/saucenao/data_source.py b/ATRI/plugins/saucenao/data_source.py index fa90e5f..c1c47d8 100644 --- a/ATRI/plugins/saucenao/data_source.py +++ b/ATRI/plugins/saucenao/data_source.py @@ -40,7 +40,7 @@ class SaouceNao(Service): res = await request.post(URL, params=self.params) except RequestError: raise RequestError("Request failed!") - data = await res.json() + data = res.json() return data async def search(self, url: str) -> str: |