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
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
|
import re
from random import choice
from nonebot.matcher import Matcher
from nonebot.params import CommandArg, ArgPlainText
from nonebot.adapters.onebot.v11 import Bot, MessageEvent, Message, MessageSegment
from nonebot.adapters.onebot.v11.helpers import (
extract_image_urls,
Cooldown,
autorevoke_send,
)
from ATRI import conf
from ATRI.service import Service
from ATRI.permission import MASTER
from .data_source import Setu
plugin = Service("涩图").document("hso!").main_cmd("/setu")
random_setu = plugin.on_command(
"来张涩图", "来张随机涩图，冷却2分钟", aliases={"涩图来", "来点涩图", "来份涩图"}, priority=5
)
@random_setu.handle([Cooldown(120)])
async def _():
...
@random_setu.handle()
async def _(bot: Bot, event: MessageEvent):
setu, setu_data = await Setu.new()
setu_info = f"Title: {setu_data.title}\nPid: {setu_data.pid}"
await bot.send(event, setu_info)
try:
await autorevoke_send(bot, event, setu)
except Exception:
await random_setu.send("hso (发不出")
await random_setu.send(f"自己动手: {setu_data.url}")
@random_setu.got("r_rush_after_think", prompt="看完不来点感想么-w-")
async def _(think: str = ArgPlainText("r_rush_after_think")):
is_repo = will_think(think)
if not is_repo:
await random_setu.finish()
else:
await random_setu.finish(is_repo)
tag_setu = plugin.on_regex(r"来[张点丶份](.*?)的?[涩色🐍]图", "根据提供的tag查找涩图，冷却2分钟", priority=6)
@tag_setu.handle([Cooldown(120)])
async def _(bot: Bot, event: MessageEvent):
msg = str(event.get_message()).strip()
pattern = r"来[张点丶份](.*?)的?[涩色🐍]图"
tag = re.findall(pattern, msg)[0]
setu, setu_data = await Setu.new(tag)
if not setu_data.url:
await tag_setu.finish("没有合适的涩图呢...")
setu_info = f"Title: {setu_data.title}\nPid: {setu_data.pid}"
await bot.send(event, setu_info)
try:
await autorevoke_send(bot, event, setu)
except Exception:
await random_setu.send("hso (发不出")
await random_setu.send(f"自己动手: {setu_data.url}")
return
@tag_setu.got("t_rush_after_think", prompt="看完不来点感想么-w-")
async def _(think: str = ArgPlainText("t_rush_after_think")):
is_repo = will_think(think)
if not is_repo:
await random_setu.finish()
else:
await random_setu.finish(is_repo)
_catcher_max_file_size = 128
_catcher_disab_gif = False
setu_catcher = plugin.on_message("涩图嗅探", "涩图嗅探器", block=False)
@setu_catcher.handle()
async def _(bot: Bot, event: MessageEvent):
args = extract_image_urls(event.message)
if not args:
return
else:
hso = list()
for i in args:
try:
data = await Setu(i).detecter(
_catcher_max_file_size, _catcher_disab_gif
)
except Exception:
return
if data > 0.7:
hso.append(data)
hso.sort(reverse=True)
if not hso:
return
elif len(hso) == 1:
u_repo = f"hso! 涩值：{'{:.2%}'.format(hso[0])}\n不行我要发给别人看"
s_repo = (
f"涩图来咧！\n{MessageSegment.image(args[0])}\n涩值: {'{:.2%}'.format(hso[0])}"
)
else:
u_repo = f"hso! 最涩的达到：{'{:.2%}'.format(hso[0])}\n不行我一定要发给别人看"
ss = list()
for s in args:
ss.append(MessageSegment.image(s))
ss = "\n".join(map(str, ss))
s_repo = f"多张涩图来咧！\n{ss}\n最涩的达到：{'{:.2%}'.format(hso[0])}"
await bot.send(event, u_repo)
for superuser in conf.BotConfig.superusers:
await bot.send_private_msg(user_id=superuser, message=s_repo)
nsfw_checker = plugin.cmd_as_group("nsfw", "涩值检测")
@nsfw_checker.got("nsfw_img", "图呢？")
async def _(bot: Bot, event: MessageEvent):
args = extract_image_urls(event.message)
if not args:
await nsfw_checker.reject("请发送图片而不是其他东西！！")
hso = await Setu(args[0]).detecter(_catcher_max_file_size, _catcher_disab_gif)
if not hso:
await nsfw_checker.finish("图不行，不测！")
resu = f"涩值：{'{:.2%}'.format(hso)}\n"
if hso >= 0.75:
resu += "hso! 不行我要发给别人看"
repo = f"涩图来咧！\n{MessageSegment.image(args[0])}\n涩值：{'{:.2%}'.format(hso)}"
for superuser in conf.BotConfig.superusers:
await bot.send_private_msg(user_id=superuser, message=repo)
elif 0.75 > hso >= 0.5:
resu += "嗯。可冲"
else:
resu += "还行8"
await nsfw_checker.finish(resu)
catcher_setting = plugin.cmd_as_group("nsfw.size", "涩图检测图片文件大小设置", permission=MASTER)
@catcher_setting.handle()
async def _(matcher: Matcher, args: Message = CommandArg()):
msg = args.extract_plain_text()
if msg:
matcher.set_arg("catcher_set", args)
@catcher_setting.got("catcher_set", "数值呢? (1对应1kb, 默认128)")
async def _(msg: str = ArgPlainText("catcher_set")):
global _catcher_max_file_size
try:
_catcher_max_file_size = int(msg)
except Exception:
await catcher_setting.reject("请发送阿拉伯数字～！")
repo = f"好诶！涩图检测文件最小值已设为：{_catcher_max_file_size}kb"
await catcher_setting.finish(repo)
animation_checker = plugin.cmd_as_group("nsfw.gif", "对动图的检测开关", permission=MASTER)
@animation_checker.handle()
async def _(event: MessageEvent):
global _catcher_disab_gif
if _catcher_disab_gif:
_catcher_disab_gif = False
else:
_catcher_disab_gif = True
await animation_checker.finish(
f"已{'禁用' if _catcher_disab_gif else '启用'}对 gif 的涩值检测"
)
_ag_l = ["涩图来", "来点涩图", "来份涩图"]
_ag_patt = r"来[张点丶份](.*?)的[涩色🐍]图"
_nice_patt = r"[hH好][sS涩色][oO哦]|[嗯恩摁社蛇🐍射]了|(硬|石更)了|[牛🐂][牛🐂]要炸了|[炼恋]起来|开?导"
_nope_patt = r"不够[涩色]|就这|不行|不彳亍|一般|这也[是叫算]|[?？]|就这|爬|爪巴"
_again_patt = r"再来一张|不够"
_nice_repo = ["w", "好诶！", "ohh", "(///w///)", "🥵", "我也"]
_nope_repo = ["那你来发", "爱看不看", "你看不看吧", "看这种类型的涩图，是一件多么美妙的事情"]
_again_repo = ["没了...", "自己找去"]
def will_think(msg: str) -> str:
if msg in _ag_l:
return str()
ag_jud = re.findall(_ag_patt, msg)
if ag_jud:
return str()
nice_jud = re.findall(_nice_patt, msg)
nope_jud = re.findall(_nope_patt, msg)
again_jud = re.findall(_again_patt, msg)
if nice_jud:
return choice(_nice_repo)
elif nope_jud:
return choice(_nope_repo)
elif again_jud:
return choice(_again_repo)
else:
return str()
|
