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
|
#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
'''
@File : body.py
@Time : 2020/10/11 14:38:23
@Author : Kyomotoi
@Contact : kyomotoiowo@gmail.com
@Github : https://github.com/Kyomotoi
@License : Copyright © 2018-2020 Kyomotoi, All Rights Reserved.
'''
__author__ = 'kyomotoi'
import json
from utils.utils_error import errorRepo
from utils.utils_request import request_get
class SauceNAO:
"""搜图请求主体"""
def __init__(self,
api_key,
output_type=2,
testmode=0,
dbmask=None,
dbmaski=32768,
db=5,
numres=1):
api = 'https://saucenao.com/search.php'
self.api = api
params = dict()
params['api_key'] = api_key
params['output_type'] = output_type
params['testmode'] = testmode
params['dbmask'] = dbmask
params['dbmaski'] = dbmaski
params['db'] = db
params['numres'] = numres
self.params = params
def search(self, url):
self.params['url'] = url
return request_get(self.api, self.params)
def resultRepo(user: str, key: str, img_url: str):
try:
task = SauceNAO(key)
data = task.search(img_url)
except Exception:
return errorRepo('请求数据失败')
data = json.loads(data)['results'][0]
msg0 = ''
print(data)
msg0 += f'[CQ:at,qq={user}]\n'
msg0 += "SauceNAO INFO:\n"
msg0 += f"[CQ:image,file={data['header'].get('thumbnail', None)}]\n"
msg0 += f"Like：{data['header'].get('similarity', 0)}%\n"
msg0 += f"Title：{data['data'].get('title', None)}\n"
msg0 += f"Pixiv ID：{data['data'].get('pixiv_id', None)}\n"
msg0 += f"Author：{data['data'].get('member_name', None)}\n"
msg0 += f"Autoor ID：{data['data'].get('member_id', None)}\n"
msg0 += f"Pixiv URL: https://www.pixiv.net/artworks/{data['data'].get('pixiv_id', None)}\n"
msg0 += f"Pic URL: https://pixiv.cat/{data['data'].get('pixiv_id', None)}.jpg"
if float(data['header'].get('similarity', 0)) < 65:
msg0 += '\n注：相似率小于65%不一定正确'
return msg0
|
