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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
import re
import os
import json
import nonebot
from pathlib import Path
from random import randint, choice
from ATRI.modules.error import errorBack
from ATRI.modules.funcControl import checkSwitch, checkNoob
from ATRI.plugins.SauceNAO import API_KEY, now_time, SauceNAO
bot = nonebot.get_bot()
@bot.on_message("group")
async def _(context):
group = context["group_id"]
user = context["user_id"]
if 0 <= now_time() < 5.5:
pass
else:
if checkNoob(user, group):
msg = str(context["message"])
print(msg)
if "搜图" in msg or "识图" in msg:
if checkSwitch("saucenao_search", group):
try:
pattern = r"CQ:reply,id=(.*?)]"
info = re.findall(pattern, msg)
msgID = info[0]
print(msgID)
except:
print('ERROR-onMessage')
return
try:
with open(Path('.') / 'ATRI' / 'data' / 'groupData' / f'{group}' / 'msgData.json', 'r') as f:
data = json.load(f)
except:
data = {}
if msgID in data.keys():
msgR = data[f"{msgID}"]["msg"]
pattern = r"url=(.*?)]"
info = re.findall(pattern, msgR)
picURL = info[0]
try:
task = SauceNAO(api_key=API_KEY)
data = task.search(url=picURL)
msg0 = ''
except:
await bot.send_msg(group_id = group, message = errorBack('请求数据失败')) # type: ignore
return
print(data)
try:
data = json.loads(data)['results'][0]
title = data['data']['title']
pixiv_id = data['data']['pixiv_id']
member_name = data['data']['member_name']
member_id = data['data']['member_id']
similarity = data['header']['similarity']
mini_url = data['header']['thumbnail']
msg0 = f'[CQ:at,qq={user}]'
msg0 += f'SauceNAO结果：'
msg0 += f'[CQ:image,file={mini_url}]\n'
msg0 += f'相似度：{similarity}%\n'
msg0 += f'标题：{title}\n'
msg0 += f'插画ID：{pixiv_id}\n'
msg0 += f'画师：{member_name}\n'
msg0 += f'画师ID：{member_id}\n'
msg0 += f'直链：https://pixiv.cat/{pixiv_id}.jpg'
except:
await bot.send_msg(group_id = group, message = errorBack('处理数据失败')) # type: ignore
return
if msg0:
if float(similarity) > 70:
await bot.send_msg(group_id = group, message = msg0) # type: ignore
else:
await bot.send_msg(group_id = group, message = 'ATRI无法找到相似的图呢...') # type: ignore
else:
await bot.send_msg(group_id = group, message = '该功能已关闭...') # type: ignore
else:
try:
with open(Path('.') / 'ATRI' / 'plugins' / 'LearnRepo' / 'LearnRepo.json', 'r') as f:
data = json.load(f)
except:
data = {}
if str(msg) in data.keys():
lt = data[f"{msg}"]
msg = lt[0]
prob = int(lt[1])
res = randint(1,prob)
if res == 1:
await bot.send_msg(
group_id = group,
message = msg
) # type: ignore
@bot.on_message('group')
async def _(context):
user = context["user_id"]
group = context["group_id"]
if checkNoob(user, group):
if 0 <= now_time() < 5.5:
pass
else:
if randint(1,20) == 4:
img = choice(
[
'11.jpg', '12.jpg', '23.jpg'
]
)
img = os.path.abspath(Path('.') / 'ATRI' / 'data' / 'emoji' / 'senren' / f'{img}')
await bot.send_msg(message = f'[CQ:image,file=file:///{img}]', auto_escape = False) # type: ignore
else:
pass
|
