1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
from random import choice
from ATRI.service import Service
from ATRI.rule import is_in_service
from ATRI.exceptions import RequestError
from ATRI.utils import request
URL = "https://saucenao.com/search.php"
__doc__ = """
以图搜图，仅限二刺螈
"""
class SaouceNao(Service):
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()
params["api_key"] = api_key
params["output_type"] = output_type
params["testmode"] = testmode
params["dbmaski"] = dbmaski
params["db"] = db
params["numres"] = numres
self.params = params
async def _request(self, url: str):
self.params["url"] = url
try:
res = await request.get(URL, params=self.params)
except RequestError:
raise RequestError("Request failed!")
data = res.json()
return data
async def search(self, url: str) -> str:
data = await self._request(url)
try:
res = data.get("results", "result")
except:
return "没有相似的结果呢..."
r = list()
for i in range(3):
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"]))
r.append(_result)
if not r:
return "没有相似的结果呢..."
msg0 = str()
for i in r:
msg0 += (
"\n——————————\n"
f"Similarity: {i['similarity']}\n"
f"Name: {i['index_name']}\n"
f"URL: {i['url'].replace('https://', '')}"
)
return msg0
|
