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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
# -*- coding:utf-8 -*-
import os
import time
import json
import sqlite3
from random import choice, randint
from pathlib import Path
import nonebot
from nonebot import on_command, CommandSession
from nonebot import session
import config # type: ignore
from ATRI.modules import response # type: ignore
bot = nonebot.get_bot()
master = config.MASTER()
apikey = bot.config.LOLICONAPI # type: ignore
URL = 'https://api.lolicon.app/setu/'
SETU_REPLY = """Title: {title}
Pid: {pid}
{setu}
---------------
Complete time:{time}s"""
setu_type = 1
@on_command('setu', patterns = (r"来[点丶张份副个幅][涩色瑟][图圖]|[涩色瑟][图圖]来|[涩色瑟][图圖][gkd|GKD|搞快点]|[gkd|GKD|搞快点][涩色瑟][图圖]|[图圖]来|[我你她他它]想要[点丶张份副][涩色瑟][图圖]|我想要[1一][张份幅副个只][涩色瑟][图圖]|[我你她他它]想[看|look][涩涩|色色]的东西"), only_to_me = False)
async def setu(session: CommandSession):
group = session.event.group_id
user = session.event.user_id
start = time.perf_counter()
with open('ATRI/plugins/noobList/noobList.json', 'r') as f:
data0 = json.load(f)
with open(Path('.') / 'ATRI' / 'plugins' / 'switch' / 'switch.json', 'r') as f:
data1 = json.load(f)
if str(user) in data0.keys():
pass
else:
if data1["setu"] == 0:
res = randint(1,10)
if 1 <= res < 9:
res = randint(1,4)
if 1 <= res < 3:
if setu_type == 1:
res = randint(1,4)
con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'nearR18.db')
cur = con.cursor()
msg = cur.execute('SELECT * FROM nearR18 ORDER BY RANDOM() limit 1;')
if 1 <= res < 3:
for i in msg:
pid = i[0]
title = i[1]
img = i[7]
end = time.perf_counter()
await session.send(
SETU_REPLY.format(
title = title,
pid = pid,
setu = img,
time = round(end - start, 3)
)
)
elif res == 4:
for i in msg:
pid = i[0]
title = i[1]
img = i[7]
end = time.perf_counter()
await session.send('我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆')
await bot.send_private_msg( # type: ignore
user_id = master,
message = f"主人，从群{group}来的涩图！热乎着！\nTitle: {title}\nPid: {pid}\n{img}\nComplete time: {round(end - start, 3)}"
)
elif setu_type == 2:
res = randint(1,4)
await session.send('别急！正在找图！')
start = time.perf_counter()
values = {
"apikey": apikey,
"r18": "0",
"num": "1"
}
try:
dc = json.loads(response.request_api_params(URL, values))
title = dc["data"][0]["title"]
pid = dc["data"][0]["pid"]
setu = dc["data"][0]["url"] #b64.b64_str_img_url(dc["data"][0]["url"])
except:
await session.send('失败了失败了失败了失...')
return
if 1 <= res < 3:
end = time.perf_counter()
await session.send(
SETU_REPLY.format(
title = title,
pid = pid,
setu = dc["data"][0]["url"],
time = round(end - start, 3)
)
)
elif res == 4:
end = time.perf_counter()
await session.send('我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆')
await bot.send_private_msg( # type: ignore
user_id = master,
message = f"主人，从群{group}来的涩图！热乎着！\nTitle: {title}\nPid: {pid}\n{setu}\nComplete time: {round(end - start, 3)}"
)
elif res == 4:
img = choice(
[
'SP.jpg', 'SP1.jpg', 'SP2.jpg'
]
)
img = Path('.') / 'ATRI' / 'data' / 'emoji' / f'{img}'
img = os.path.abspath(img)
await session.send(f'[CQ:image,file=file:///{img}]')
elif res == 10:
img = choice(
[
'GDZ.png', 'SHZY1.jpg', 'SHZY2.jpg', 'SHZY3.jpg', 'SHZY4.jpg', 'SHZY5.jpg', 'SHZY6.jpg'
]
)
img = Path('.') / 'ATRI' / 'data' / 'img' / 'niceIMG' / f'{img}'
await session.send(f'[CQ:image,file=file:///{img}]')
else:
await session.send('该功能已被禁用...')
@on_command('change_setu_type', aliases = ['涩图导向'], only_to_me =False)
async def _(session: CommandSession):
global setu_type
if session.event.user_id == master:
msg = session.event.raw_message.split(' ', 1)
s_type = msg[1]
if s_type == '数据库':
setu_type = 1
elif s_type == '接口':
setu_type = 2
else:
pass
await session.send('okay~~~~')
|
