From c8994250f49e747e666367ecec1fec60e9e2b376 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Wed, 9 Feb 2022 15:33:50 +0800 Subject: =?UTF-8?q?=F0=9F=90=9B=20=E4=BF=AE=E5=A4=8D=E6=90=9C=E5=9B=BE?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=A4=84=E7=90=86=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/saucenao/data_source.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/ATRI/plugins/saucenao/data_source.py b/ATRI/plugins/saucenao/data_source.py index 08ba64e..2510cb0 100644 --- a/ATRI/plugins/saucenao/data_source.py +++ b/ATRI/plugins/saucenao/data_source.py @@ -36,7 +36,7 @@ class SaouceNao(Service): async def _request(self, url: str): self.params["url"] = url try: - res = await request.post(URL, params=self.params) + res = await request.get(URL, params=self.params) except RequestError: raise RequestError("Request failed!") data = res.json() @@ -45,24 +45,27 @@ class SaouceNao(Service): async def search(self, url: str) -> str: data = await self._request(url) try: - res = data["result"] + res = data.get("results", "result") except: return "没有相似的结果呢..." - result = list() + r = list() for i in range(3): - sim = res[i]["header"]["similarity"] - if float(sim) >= 70: - data = res[i] + data = res[i] + sim = data["header"]["similarity"] + if float(sim) >= 70: _result = dict() _result["similarity"] = sim _result["index_name"] = data["header"]["index_name"] _result["url"] = choice(data["data"].get("ext_urls", ["None"])) - result.append(_result) + r.append(_result) + + if not r: + return "没有相似的结果呢..." msg0 = str() - for i in result: + for i in r: msg0 += ( "\n——————————\n" f"Similarity: {i['similarity']}\n" -- cgit v1.2.3