diff options
-rw-r--r-- | ATRI/plugins/saucenao/data_source.py | 19 |
1 files 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" |