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
|
import json
from typing import List
from ATRI.utils import request
from ATRI.message import MessageBuilder
from .models import SauceNAORequest, SauceNAOResponse, SauceNAOResult
SAUCENAO_URL: str = "https://saucenao.com/search.php"
class SauceNAO:
def __init__(
self,
api_key: str,
output_type: int = 2,
testmode: int = 1,
dbmaski: int = 32768,
db: int = 5,
numres: int = 5,
) -> None:
self.params = SauceNAORequest(
api_key=api_key,
output_type=output_type,
testmode=testmode,
dbmaski=dbmaski,
db=db,
numres=numres,
)
async def _request(self, url: str) -> SauceNAOResponse:
self.params.url = url
resp = await request.get(SAUCENAO_URL, params=self.params.dict())
return SauceNAOResponse.parse_obj(resp.json())
async def search(self, url: str) -> str:
try:
data = await self._request(url)
except Exception as err:
raise Exception(f"处理 SauceNAO 数据失败：{str(err)}")
r: List[SauceNAOResult] = list()
for i in range(3):
_data = data.results[i]
sim = _data.header.similarity
if float(sim) >= 70:
r.append(
SauceNAOResult(
similarity=sim,
index_name=_data.header.index_name,
url=_data.data.ext_urls[0] if _data.data.ext_urls else "None",
)
)
if not r:
return "SauceNAO 中没有相似的结果"
result = str()
for i in r:
result += (
MessageBuilder("\n——————————")
.text(f"相似度：{i.similarity}")
.text(f"名称：{i.index_name}")
.text(f"URL: {i.url.replace('https://', str()).replace('http://', str())}")
.done()
)
return result
|
