async def _(session: CommandSession):
song_name = session.current_arg_text.strip()
if not song_name:
song_name = session.get('message', prompt='请告诉ATRI要点的歌曲名~')
await session.send('在找了在找了……')
URL = url + song_name
resp = response.request_api_params(URL, para)
if not resp:
session.finish('ATRI在网络上走散了...请重试...')
else:
resu = json.loads(resp)
id = resu['result']['songs'][0]['id']
msg = 'https://music.163.com/#/song?id=' + str(id)
await session.send(msg)
async def setu(session: CommandSession):
with open('ATRI/plugins/switch/switch.json', 'r') as f:
data = json.load(f)
if data["setu"] == 0:
await session.send('别急！正在找图！')
start = time.perf_counter()
values = {"apikey": apikey, "r18": "0", "num": "1"}
dc = json.loads(response.request_api_params(URL, values))
end = time.perf_counter()
await session.send(
SETU_REPLY.format(title=dc["data"][0]["title"],
pid=dc["data"][0]["pid"],
setu=dc["data"][0]["url"],
time=round(end - start, 3)))
else:
await session.send('该功能已被禁用...')
async def setu(session: CommandSession):
start = time.perf_counter()
user = session.event.user_id
group = session.event.group_id
try:
with open(
Path('.') / 'ATRI' / 'plugins' / 'noobList' / 'noobGroup.json',
'r') as f:
data = json.load(f)
except:
data = {}
try:
with open(
Path('.') / 'ATRI' / 'plugins' / 'noobList' / 'noobList.json',
'r') as f:
data1 = json.load(f)
except:
data1 = {}
if str(group) in data.keys():
pass
else:
if str(user) in data1.keys():
pass
else:
if 0 <= now_time() < 5.5:
await session.send(
choice([
'zzzz......', 'zzzzzzzz......', 'zzz...好涩哦..zzz....',
'别...不要..zzz..那..zzz..', '嘻嘻..zzz..呐~..zzzz..'
]))
else:
with open(
Path('.') / 'ATRI' / 'plugins' / 'switch' /
'switch.json', 'r') as f:
data1 = json.load(f)
if data1["setu"] == "on":
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_LOLI,
"r18": "0",
"num": "1"
}
try:
dc = json.loads(
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}'
img = os.path.abspath(img)
await session.send(f'[CQ:image,file=file:///{img}]')
else:
await session.send('该功能已被禁用...')
async def _(context):
start = time.perf_counter()
user = context["user_id"]
group = context["group_id"]
if 0 <= now_time() < 5.5:
pass
else:
try:
with open(
Path('.') / 'ATRI' / 'plugins' / 'noobList' /
'noobGroup.json', 'r') as f:
data = json.load(f)
except:
data = {}
try:
with open(
Path('.') / 'ATRI' / 'plugins' / 'noobList' /
'noobList.json', 'r') as f:
data1 = json.load(f)
except:
data1 = {}
if str(group) in data.keys():
pass
else:
if str(user) in data1.keys():
pass
else:
try:
img = str(context["message"])
pattern = re.compile(r"url=(.*)]")
img = re.findall(pattern, img)
img = img[0].replace('url=', '')
img = img.replace(']', '')
print(img)
except:
return
img = b64_str_img_url(img)
if img:
host = f'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={API_KEY}&client_secret={SECRECT_KEY}'
headers = {
'Content-Type': 'application/json;charset=UTF-8'
}
res = json.loads(request_api_params(host, headers))
access_token = res['access_token']
url = f'https://aip.baidubce.com/rest/2.0/ocr/v1/general?access_token={access_token}'
headers = {
'Content-Type': 'application/x-www-form-urlencoded'
}
data = urlencode({'image': img})
res = requests.post(url=url, headers=headers, data=data)
try:
words = json.loads(
res.content)['words_result'][0]['words']
print(words)
except:
return
if re.findall(r"[涩色]图", words):
with open(
Path('.') / 'ATRI' / 'plugins' / 'switch' /
'switch.json', 'r') as f:
data1 = json.load(f)
if data1["setu"] == "on":
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()
msg = SETU_REPLY.format(
title=title,
pid=pid,
setu=img,
time=round(end - start, 3))
await bot.send_group_msg(
group_id=group,
message=msg
) # type: ignore
elif res == 4:
for i in msg:
pid = i[0]
title = i[1]
img = i[7]
end = time.perf_counter()
await bot.send_group_msg(
group_id=group,
message=
'我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆'
) # type: ignore
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 bot.send_group_msg(
group_id=group,
message='别急！正在找图！') # type: ignore
start = time.perf_counter()
values = {
"apikey": apikey_LOLI,
"r18": "0",
"num": "1"
}
try:
dc = json.loads(
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 bot.send_group_msg(
group_id=group,
message='失败了失败了失...'
) # type: ignore
return
if 1 <= res < 3:
end = time.perf_counter()
msg = SETU_REPLY.format(
title=title,
pid=pid,
setu=img,
time=round(end - start, 3))
await bot.send_group_msg(
group_id=group,
message=msg) # type: ignore
elif res == 4:
end = time.perf_counter()
await bot.send_group_msg(
group_id=group,
message=
'我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆'
) # type: ignore
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 bot.send_group_msg(
group_id=group,
message=f'[CQ:image,file=file:///{img}]'
) # type: ignore
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}'
img = os.path.abspath(img)
await bot.send_group_msg(
group_id=group,
message=f'[CQ:image,file=file:///{img}]'
) # type: ignore
else:
pass
async def setu(session: CommandSession):
start = time.perf_counter()
user = session.event.user_id
group = session.event.group_id
if checkNoob(user, group):
if 0 <= now_time() < 5.5:
await session.send(
choice([
'zzzz......', 'zzzzzzzz......', 'zzz...好涩哦..zzz....',
'别...不要..zzz..那..zzz..', '嘻嘻..zzz..呐~..zzzz..'
]))
else:
if checkSwitch(__plugin_name__, group):
res = randint(1, 10)
if 1 <= res <= 9:
res = randint(1, 4)
if 1 <= res <= 3:
if setu_type == 1:
res = randint(1, 5)
await session.send('别急！正在找图！')
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 <= 4:
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 == 5:
for i in msg:
pid = i[0]
title = i[1]
img = i[7]
end = time.perf_counter()
await session.send(
'我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆')
await send_to_superusers(
bot,
message=
f"主人，从群{group}来的涩图！热乎着！\nTitle: {title}\nPid: {pid}\n{img}\nComplete time: {round(end - start, 3)}"
)
elif setu_type == 2:
res = randint(1, 5)
await session.send('别急！正在找图！')
start = time.perf_counter()
values = {
"apikey": apikey_LOLI,
"r18": "0",
"num": "1"
}
try:
dc = json.loads(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:
session.finish(errorBack('请求数据失败'))
if 1 <= res <= 4:
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 == 5:
end = time.perf_counter()
await session.send(
'我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆')
await send_to_superusers(
bot,
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}'
img = os.path.abspath(img)
await session.send(f'[CQ:image,file=file:///{img}]')
else:
await session.send('该功能已关闭...')
async def _(context):
start = time.perf_counter()
user = context["user_id"]
group = context["group_id"]
if checkNoob(user, group):
if checkSwitch(__plugin_name1__, group):
try:
img = str(context["message"])
pattern = re.compile(r"url=(.*)]")
img = re.findall(pattern, img)
img = img[0].replace('url=', '')
img = img.replace(']', '')
print(img)
except:
return
try:
img = b64_str_img_url(img)
print('转换图片至base64成功')
except:
return
try:
host = f'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={API_KEY}&client_secret={SECRECT_KEY}'
headers = {'Content-Type': 'application/json;charset=UTF-8'}
res = json.loads(request_api_params(host, headers))
access_token = res['access_token']
url = f'https://aip.baidubce.com/rest/2.0/ocr/v1/general?access_token={access_token}'
headers = {'Content-Type': 'application/x-www-form-urlencoded'}
data = urlencode({'image': img})
# res = requests.post(url=url, headers=headers, data=data)
async def func0(url, headers, data):
async with aiohttp.ClientSession() as client:
async with client.post(url, headers=headers,
data=data) as req:
res = await req.read()
return res
words = json.loads(str(func0(
url, headers, data)))['words_result'][0]['words']
print('BaiduAPI请求成功')
except:
return
if re.findall(r"[涩色]图|炼铜", words):
if checkSwitch(__plugin_name__, group):
res = randint(1, 10)
if 1 <= res <= 9:
res = randint(1, 5)
if 1 <= res <= 4:
if setu_type == 1:
res = randint(1, 5)
await bot.send_group_msg(group_id=group,
message='别急！正在找图！'
) # type: ignore
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 <= 4:
for i in msg:
pid = i[0]
title = i[1]
img = i[7]
end = time.perf_counter()
msg = SETU_REPLY.format(
title=title,
pid=pid,
setu=img,
time=round(end - start, 3))
await bot.send_group_msg(
group_id=group,
message=msg) # type: ignore
elif res == 5:
for i in msg:
pid = i[0]
title = i[1]
img = i[7]
end = time.perf_counter()
await bot.send_group_msg(
group_id=group,
message='我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆'
) # type: ignore
await send_to_superusers(
bot,
message=
f"主人，从群{group}来的涩图！热乎着！\nTitle: {title}\nPid: {pid}\n{img}\nComplete time: {round(end - start, 3)}"
)
elif setu_type == 2:
res = randint(1, 5)
await bot.send_group_msg(group_id=group,
message='别急！正在找图！'
) # type: ignore
start = time.perf_counter()
values = {
"apikey": apikey_LOLI,
"r18": "0",
"num": "1"
}
try:
dc = json.loads(
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 bot.send_group_msg(
group_id=group,
message=errorBack('数据请求失败')
) # type: ignore
return
if 1 <= res <= 4:
end = time.perf_counter()
msg = SETU_REPLY.format(title=title,
pid=pid,
setu=img,
time=round(
end - start,
3))
await bot.send_group_msg(group_id=group,
message=msg
) # type: ignore
elif res == 5:
end = time.perf_counter()
await bot.send_group_msg(
group_id=group,
message='我找到涩图了！但我发给主人了\nο(=•ω＜=)ρ⌒☆'
) # type: ignore
await send_to_superusers(
bot,
message=
f"主人，从群{group}来的涩图！热乎着！\nTitle: {title}\nPid: {pid}\n{setu}\nComplete time: {round(end - start, 3)}"
)
elif res == 5:
img = choice(['SP.jpg', 'SP1.jpg', 'SP2.jpg'])
img = Path(
'.') / 'ATRI' / 'data' / 'emoji' / f'{img}'
img = os.path.abspath(img)
await bot.send_group_msg(
group_id=group,
message=f'[CQ:image,file=file:///{img}]'
) # type: ignore
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}'
img = os.path.abspath(img)
await bot.send_group_msg(
group_id=group,
message=f'[CQ:image,file=file:///{img}]'
) # type: ignore
else:
pass
else:
pass