diff options
author | Lint Action <[email protected]> | 2021-08-10 12:03:23 +0000 |
---|---|---|
committer | Lint Action <[email protected]> | 2021-08-10 12:03:23 +0000 |
commit | 07a4e75c7d0c1e300e3f813c75b12b43bfd7fdd1 (patch) | |
tree | 05a48771d6bcc35f8731855c9204423a839afa5d /ATRI/plugins/saucenao | |
parent | 0070a3efef1c7e6165f62ad39a9be53b65b12a29 (diff) | |
download | ATRI-07a4e75c7d0c1e300e3f813c75b12b43bfd7fdd1.tar.gz ATRI-07a4e75c7d0c1e300e3f813c75b12b43bfd7fdd1.tar.bz2 ATRI-07a4e75c7d0c1e300e3f813c75b12b43bfd7fdd1.zip |
:rotating_light: 自动进行代码格式化
Diffstat (limited to 'ATRI/plugins/saucenao')
-rw-r--r-- | ATRI/plugins/saucenao/data_source.py | 25 |
1 files changed, 16 insertions, 9 deletions
diff --git a/ATRI/plugins/saucenao/data_source.py b/ATRI/plugins/saucenao/data_source.py index 58b031b..3f77118 100644 --- a/ATRI/plugins/saucenao/data_source.py +++ b/ATRI/plugins/saucenao/data_source.py @@ -15,8 +15,15 @@ __doc__ = """ class SaouceNao(Service): - - def __init__(self, api_key: str = None, output_type=2, testmode=1, dbmaski=32768, db=5, numres=5): + def __init__( + self, + api_key: str = None, + output_type=2, + testmode=1, + dbmaski=32768, + db=5, + numres=5, + ): Service.__init__(self, "以图搜图", __doc__, rule=is_in_service("以图搜图")) params = dict() @@ -27,31 +34,31 @@ class SaouceNao(Service): params["db"] = db params["numres"] = numres self.params = params - + async def _request(self, url: str): self.params["url"] = url - + try: res = await request.post(URL, params=self.params) except RequestError: raise RequestError("Request failed!") data = await res.json() return data - + async def search(self, url: str) -> str: data = await self._request(url) res = data["results"] - + result = list() for i in range(len(res)): 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.append(_result) - + msg0 = str() for i in result: msg0 += ( @@ -60,7 +67,7 @@ class SaouceNao(Service): f"Name: {i['index_name']}\n" f"URL: {i['url'].replace('https://', '')}" ) - + if len(res) <= 3: return msg0 else: |