summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2022-02-09 15:33:50 +0800
committerKyomotoi <[email protected]>2022-02-09 15:33:50 +0800
commitc8994250f49e747e666367ecec1fec60e9e2b376 (patch)
tree9a676d2e1b0e3aae5b30efced83465bd18e2377b
parent2c3027c85018dc7aa123b87b5b5144eb1cc67c65 (diff)
downloadATRI-c8994250f49e747e666367ecec1fec60e9e2b376.tar.gz
ATRI-c8994250f49e747e666367ecec1fec60e9e2b376.tar.bz2
ATRI-c8994250f49e747e666367ecec1fec60e9e2b376.zip
šŸ› äæ®å¤ęœå›¾ę•°ę®å¤„ē†é”™čÆÆ
-rw-r--r--ATRI/plugins/saucenao/data_source.py19
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"