mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-30 09:10:01 +08:00
first_commit .gitignore
This commit is contained in:
hibiki
@@ -0,0 +1,276 @@
|
||||
import random
|
||||
from nonebot import on_command, on_regex
|
||||
from services.log import logger
|
||||
from models.sigin_group_user import SignGroupUser
|
||||
from util.utils import FreqLimiter, UserExistLimiter, is_number, get_message_text, get_message_imgs
|
||||
from nonebot.typing import T_State
|
||||
from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent, PrivateMessageEvent
|
||||
from .data_source import get_setu, get_luoxiang, search_online_setu, get_setu_urls, \
|
||||
check_r18_and_keyword, find_img_index
|
||||
from nonebot.adapters.cqhttp.exception import ActionFailed
|
||||
import re
|
||||
from models.count_user import UserCount
|
||||
from aiohttp.client_exceptions import ClientConnectorError
|
||||
from configs.config import LOCAL_SETU
|
||||
|
||||
__plugin_name__ = '色图'
|
||||
__plugin_usage__ = '''示例：
|
||||
1. 色图 （随机本地色图）
|
||||
2. 色图r （随机在线十张r18涩图）
|
||||
3. 色图 666 （本地色图id）
|
||||
4. 色图 xx （在线搜索xx色图）
|
||||
5. 色图r xx （搜索十张xx的r18涩图，注意空格）（仅私聊）
|
||||
6. 来n张涩图 （本地涩图连发）（1<=n<=9）
|
||||
7. 来n张xx的涩图 （在线搜索xx涩图）（较慢，看网速）
|
||||
注：【色图r每日提供5次
|
||||
本地涩图没有r18！
|
||||
联网搜索会较慢！
|
||||
如果图片数量与数字不符，
|
||||
原因1：网络不好，网线被拔QAQ
|
||||
原因2：搜索到的总数小于数字
|
||||
原因3：图太色或者小错误了】'''
|
||||
|
||||
url = "https://api.lolicon.app/setu/"
|
||||
_flmt = FreqLimiter(5)
|
||||
_ulmt = UserExistLimiter()
|
||||
path = "setu/"
|
||||
MAX_COUNT = 5
|
||||
|
||||
|
||||
setu = on_command("色图", aliases={"涩图", "不够色", "来一发", "再来点"}, priority=5, block=True)
|
||||
setu_reg = on_regex('(.*)[份|发|张|个|次|点](.*)[瑟|色|涩]图', priority=5, block=True)
|
||||
find_setu = on_command("查色图", priority=5, block=True)
|
||||
|
||||
|
||||
@setu.handle()
|
||||
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||
img_id = get_message_text(event.json())
|
||||
if img_id in ['帮助']:
|
||||
await setu.finish(__plugin_usage__)
|
||||
impression = (await SignGroupUser.ensure(event.user_id, event.group_id)).impression
|
||||
luox = get_luoxiang(impression)
|
||||
if luox:
|
||||
await setu.finish(luox)
|
||||
if not _flmt.check(event.user_id):
|
||||
await setu.finish('您冲得太快了，请稍候再冲', at_sender=True)
|
||||
_flmt.start_cd(event.user_id)
|
||||
if _ulmt.check(event.user_id):
|
||||
await setu.finish(f"您有色图正在处理，请稍等")
|
||||
_ulmt.set_True(event.user_id)
|
||||
setu_img, index = get_setu(img_id)
|
||||
if setu_img and LOCAL_SETU:
|
||||
try:
|
||||
await setu.send(setu_img)
|
||||
except:
|
||||
_ulmt.set_False(event.user_id)
|
||||
await setu.finish('这张图色过头了，我自己看看就行了！', at_sender=True)
|
||||
logger.info(
|
||||
f"USER {event.user_id} GROUP {event.group_id} 发送色图 {index}.jpg 成功")
|
||||
else:
|
||||
msg = img_id
|
||||
if msg.find(list(bot.config.nickname)[0]) != -1:
|
||||
_ulmt.set_False(event.user_id)
|
||||
await setu.finish('咳咳咳，虽然我很可爱，但是我木有自己的色图~~~有的话记得发我一份呀')
|
||||
keyword, r18, num = await check_r18_and_keyword(msg, event.user_id)
|
||||
if r18 == 1:
|
||||
_ulmt.set_False(event.user_id)
|
||||
await setu.finish(random.choice([
|
||||
"这种不好意思的东西怎么可能给这么多人看啦",
|
||||
"羞羞脸！给我滚出克私聊！",
|
||||
"变态变态变态变态大变态！"
|
||||
]))
|
||||
try:
|
||||
urls, text_list, code = await get_setu_urls(keyword, num, r18=r18)
|
||||
except ClientConnectorError:
|
||||
await setu.finish('网络失败了..别担心！正在靠运气上网！', at_sender=True)
|
||||
else:
|
||||
if code == 200:
|
||||
for i in range(num):
|
||||
try:
|
||||
setu_img, index = await search_online_setu(urls[i])
|
||||
await setu.send(text_list[i] + setu_img)
|
||||
logger.info(
|
||||
f"USER {event.user_id} GROUP {event.group_id}"
|
||||
f" 发送在线色图 {keyword}.jpg 成功")
|
||||
except Exception as e:
|
||||
logger.error(f'色图发送错误 e：{e}')
|
||||
await setu.send('图片下载惜败！', at_sender=True)
|
||||
else:
|
||||
await setu.send(urls)
|
||||
_ulmt.set_False(event.user_id)
|
||||
|
||||
|
||||
@setu.handle()
|
||||
async def _(bot: Bot, event: PrivateMessageEvent, state: T_State):
|
||||
img_id = get_message_text(event.json())
|
||||
if img_id in ['帮助']:
|
||||
await setu.finish(__plugin_usage__)
|
||||
if not _flmt.check(event.user_id):
|
||||
await setu.finish('您冲得太快了，请稍候再冲', at_sender=True)
|
||||
_flmt.start_cd(event.user_id)
|
||||
if _ulmt.check(event.user_id):
|
||||
await setu.finish(f"您有色图正在处理，请稍等")
|
||||
_ulmt.set_True(event.user_id)
|
||||
setu_img, index = get_setu(img_id)
|
||||
if setu_img:
|
||||
await setu.send(setu_img)
|
||||
logger.info(
|
||||
f"USER {event.user_id} GROUP private 发送色图 {index}.jpg 成功")
|
||||
else:
|
||||
msg = img_id
|
||||
if msg.find(list(bot.config.nickname)[0]) != -1:
|
||||
_ulmt.set_False(event.user_id)
|
||||
await setu.finish('咳咳咳，虽然我很可爱，但是我木有自己的色图~~~有的话记得发我一份呀')
|
||||
keyword, r18, num = await check_r18_and_keyword(msg, event.user_id)
|
||||
if r18 == 1:
|
||||
if await UserCount.check_count(event.user_id, 'setu_r18', MAX_COUNT):
|
||||
_ulmt.set_False(event.user_id)
|
||||
await setu.finish('要节制啊，请明天再来...\n【每日提供5次】', at_sender=True)
|
||||
try:
|
||||
urls, text_list, code = await get_setu_urls(keyword, num, r18=r18)
|
||||
except ClientConnectorError:
|
||||
await UserCount.add_count(event.user_id, 'setu_r18', count=-1)
|
||||
await setu.finish('网络失败了..别担心！这次搜索不算数喔', at_sender=True)
|
||||
else:
|
||||
count = 0
|
||||
if code == 200:
|
||||
for i in range(num):
|
||||
try:
|
||||
setu_img, index = await search_online_setu(urls[i])
|
||||
await setu.send(text_list[i] + setu_img)
|
||||
logger.info(
|
||||
f"USER {event.user_id} GROUP private"
|
||||
f" 发送{'r18' if img_id == 'r' else ''}色图 {index}.jpg 成功")
|
||||
except Exception as e:
|
||||
logger.error(f'色图发送错误 e：{e}')
|
||||
await setu.send('图片下载惜败！', at_sender=True)
|
||||
count += 1
|
||||
if count > 6:
|
||||
await setu.send('检测到下载惜败的图片过多，这次就不算数了，果咩..', at_sender=True)
|
||||
await UserCount.add_count(event.user_id, 'setu_r18', count=-1)
|
||||
else:
|
||||
if code == 401:
|
||||
if r18 == 1:
|
||||
await UserCount.add_count(event.user_id, 'setu_r18', count=-1)
|
||||
await setu.send(urls + ' 色图r次数返还！')
|
||||
else:
|
||||
await setu.send(urls, at_sender=True)
|
||||
else:
|
||||
if r18 == 1:
|
||||
await setu.send('这次不是小真寻的戳！色图r次数返还！', at_sender=True)
|
||||
await UserCount.add_count(event.user_id, 'setu_r18', count=-1)
|
||||
else:
|
||||
await setu.send(urls, at_sender=True)
|
||||
_ulmt.set_False(event.user_id)
|
||||
|
||||
|
||||
num_key = {
|
||||
'一': 1,
|
||||
'二': 2,
|
||||
'两': 2,
|
||||
'双': 2,
|
||||
'三': 3,
|
||||
'四': 4,
|
||||
'五': 5,
|
||||
'六': 6,
|
||||
'七': 7,
|
||||
'八': 8,
|
||||
'九': 9
|
||||
}
|
||||
|
||||
|
||||
@setu_reg.handle()
|
||||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
if event.message_type == "group":
|
||||
impression = (await SignGroupUser.ensure(event.user_id, event.group_id)).impression
|
||||
luox = get_luoxiang(impression)
|
||||
if luox:
|
||||
await setu.finish(luox, at_sender=True)
|
||||
if _ulmt.check(event.user_id):
|
||||
await setu.finish(f"您有色图正在处理，请稍等")
|
||||
_ulmt.set_True(event.user_id)
|
||||
if not _flmt.check(event.user_id):
|
||||
_ulmt.set_False(event.user_id)
|
||||
await setu.finish('您冲得太快了，请稍候再冲', at_sender=True)
|
||||
_flmt.start_cd(event.user_id)
|
||||
msg = get_message_text(event.json())
|
||||
num = 1
|
||||
msg = re.search(r'(.*)[份|发|张|个|次|点](.*)[瑟|涩|色]图', msg)
|
||||
if msg:
|
||||
num = msg.group(1)
|
||||
keyword = msg.group(2)
|
||||
if keyword:
|
||||
if keyword[-1] == '的':
|
||||
keyword = keyword[:-1]
|
||||
if num:
|
||||
num = num[-1]
|
||||
if num_key.get(num):
|
||||
num = num_key[num]
|
||||
elif is_number(num):
|
||||
try:
|
||||
num = int(num)
|
||||
except ValueError:
|
||||
num = 1
|
||||
else:
|
||||
num = 1
|
||||
else:
|
||||
return
|
||||
# try:
|
||||
if not keyword:
|
||||
for _ in range(num):
|
||||
try:
|
||||
img, index = get_setu('')
|
||||
await setu_reg.send(img)
|
||||
except Exception as e:
|
||||
await setu_reg.send('有图太色了发不出来...')
|
||||
else:
|
||||
logger.info(
|
||||
f"USER {event.user_id} GROUP {event.group_id if event.message_type != 'private' else 'private'}"
|
||||
f" 发送 {index} 色图成功")
|
||||
else:
|
||||
urls, text_list, code = await get_setu_urls(keyword, num)
|
||||
if code == 200:
|
||||
for i in range(len(urls)):
|
||||
try:
|
||||
setu_img, index = await search_online_setu(urls[i])
|
||||
await setu_reg.send(text_list[i] + '\n' + setu_img)
|
||||
except ActionFailed as e:
|
||||
await setu_reg.send('这图太色了，会教坏小孩子的，不给看..')
|
||||
else:
|
||||
logger.info(
|
||||
f"USER {event.user_id} GROUP {event.group_id if event.message_type != 'private' else 'private'}"
|
||||
f" 发送 {keyword} {num}连 色图成功")
|
||||
else:
|
||||
_ulmt.set_False(event.user_id)
|
||||
await setu_reg.finish(urls, at_sender=True)
|
||||
_ulmt.set_False(event.user_id)
|
||||
|
||||
|
||||
@find_setu.args_parser
|
||||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
if str(event.message) == '取消':
|
||||
await find_setu.finish('取消了操作', at_sender=True)
|
||||
imgs = get_message_imgs(event.json())
|
||||
if not imgs:
|
||||
await find_setu.reject("不搞错了，俺要图！")
|
||||
state['img'] = imgs[0]
|
||||
|
||||
|
||||
@find_setu.handle()
|
||||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
if get_message_text(event.json()) in ['帮助']:
|
||||
await find_setu.finish('通过图片获取本地色图id\n\t示例：查色图(图片)')
|
||||
imgs = get_message_imgs(event.json())
|
||||
if imgs:
|
||||
state['img'] = imgs[0]
|
||||
|
||||
|
||||
@find_setu.got('img', prompt="速速来图！")
|
||||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
img = state['img']
|
||||
await find_setu.send(await find_img_index(img, event.user_id), at_sender=True)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,62 @@
|
||||
import os
|
||||
import imagehash
|
||||
from PIL import Image
|
||||
try:
|
||||
import ujson as json
|
||||
except ModuleNotFoundError:
|
||||
import json
|
||||
|
||||
IMAGE_PATH = r"/home/hibiki/hibikibot/resources/img/"
|
||||
TXT_PATH = r"/home/hibiki/hibikibot/resources/txt/"
|
||||
|
||||
|
||||
def get_img_hash(image_file):
|
||||
with open(image_file, 'rb') as fp:
|
||||
hash_value = imagehash.average_hash(Image.open(fp))
|
||||
return hash_value
|
||||
|
||||
|
||||
def check_file_index():
|
||||
for dir_name in ['setu/', 'r18/']:
|
||||
lens = len(os.listdir(IMAGE_PATH + dir_name))
|
||||
for i in range(lens):
|
||||
if i > lens:
|
||||
return
|
||||
if not os.path.exists(f"{IMAGE_PATH}{dir_name}{i}.jpg"):
|
||||
os.rename(f"{IMAGE_PATH}{dir_name}{lens}.jpg", f"{IMAGE_PATH}{dir_name}{i}.jpg")
|
||||
print(f'{lens}.jpg --> {i}.jpg')
|
||||
lens -= 1
|
||||
if not i % 100:
|
||||
print(f'已检测 {i} 份数据')
|
||||
print(f'{dir_name} 检测完毕')
|
||||
|
||||
|
||||
def check_setu_hash():
|
||||
check_file_index()
|
||||
for dir_name in ['setu/', 'r18/']:
|
||||
img_data = {}
|
||||
if dir_name == 'setu/':
|
||||
fn = 'setu_img_hash.json'
|
||||
else:
|
||||
fn = 'r18_setu_img_hash.json'
|
||||
file_list_len = len(os.listdir(IMAGE_PATH + dir_name)) - 1
|
||||
print(file_list_len)
|
||||
for i in range(file_list_len):
|
||||
file = f"{i}.jpg"
|
||||
index = file.split(".")[0]
|
||||
img_hash = str(get_img_hash(IMAGE_PATH + dir_name + file))
|
||||
print(f'{index}.jpg --> {img_hash}')
|
||||
if img_hash in img_data.values():
|
||||
k = [k for k, v in img_data.items() if v == img_hash]
|
||||
print(f'文件 {index}.jpg 与 {k}.jpg 重复，使用 {file_list_len}.jpg 进行替换')
|
||||
os.remove(IMAGE_PATH + dir_name + file)
|
||||
os.rename(IMAGE_PATH + f'{dir_name}{file_list_len}.jpg', IMAGE_PATH + dir_name + file)
|
||||
file_list_len -= 1
|
||||
continue
|
||||
img_data[index] = img_hash
|
||||
# print(f'{index}.jpg --> {img_data}')
|
||||
with open(TXT_PATH + fn, 'w') as f:
|
||||
json.dump(img_data, f, indent=4)
|
||||
|
||||
|
||||
check_setu_hash()
|
||||
@@ -0,0 +1,168 @@
|
||||
from configs.path_config import IMAGE_PATH, TXT_PATH
|
||||
import os
|
||||
import random
|
||||
from util.init_result import image
|
||||
from configs.config import LOLICON_KEY
|
||||
import aiohttp
|
||||
import aiofiles
|
||||
from services.log import logger
|
||||
from util.img_utils import get_img_hash
|
||||
from util.utils import get_local_proxy, is_number
|
||||
from asyncio.exceptions import TimeoutError
|
||||
from models.count_user import UserCount
|
||||
from configs.config import DOWNLOAD_SETU
|
||||
try:
|
||||
import ujson as json
|
||||
except ModuleNotFoundError:
|
||||
import json
|
||||
|
||||
|
||||
url = "https://api.lolicon.app/setu/"
|
||||
path = 'setu/'
|
||||
|
||||
|
||||
async def get_setu_urls(keyword: str, num: int = 1, r18: int = 0):
|
||||
# print(keyword)
|
||||
if r18 == 1:
|
||||
file_name = 'setu_r18_url.txt'
|
||||
else:
|
||||
file_name = 'setu_url.txt'
|
||||
try:
|
||||
with open(TXT_PATH + file_name, 'r') as f:
|
||||
txt_data = f.read()
|
||||
except FileNotFoundError:
|
||||
txt_data = ''
|
||||
params = {
|
||||
"apikey": LOLICON_KEY, # 添加apikey
|
||||
'r18': r18, # 添加r18参数 0为否，1为是，2为混合
|
||||
'keyword': keyword, # 若指定关键字，将会返回从插画标题、作者、标签中模糊搜索的结果
|
||||
'num': num, # 一次返回的结果数量，范围为1到10，不提供 APIKEY 时固定为1
|
||||
'size1200': 1, # 是否使用 master_1200 缩略图，以节省流量或提升加载速度
|
||||
}
|
||||
urls = []
|
||||
text_list = []
|
||||
for count in range(3):
|
||||
print(f'get_setu_url: count --> {count}')
|
||||
async with aiohttp.ClientSession() as session:
|
||||
try:
|
||||
async with session.get(url, proxy=get_local_proxy(), timeout=5, params=params) as response:
|
||||
if response.status == 429:
|
||||
return '调用达到上限，明日赶早呀~', '', 429
|
||||
if response.status == 404:
|
||||
return "网站裂开了...", '', 998
|
||||
if response.status == 200:
|
||||
data = await response.json()
|
||||
if data['code'] == 0:
|
||||
# print(len(data['data']))
|
||||
for i in range(len(data['data'])):
|
||||
img_url = data['data'][i]['url']
|
||||
title = data['data'][i]['title']
|
||||
author = data['data'][i]['author']
|
||||
pid = data['data'][i]['pid']
|
||||
urls.append(img_url)
|
||||
text_list.append(f'title：{title}\nauthor：{author}\nPID：{pid}')
|
||||
img_url = str(img_url).replace('img-master', 'img-original').replace('_master1200', '')
|
||||
txt_data += img_url + ','
|
||||
if DOWNLOAD_SETU:
|
||||
with open(TXT_PATH + file_name, 'w') as f:
|
||||
f.write(txt_data)
|
||||
return urls, text_list, 200
|
||||
else:
|
||||
return "没找到符合条件的色图...", '', 401
|
||||
except TimeoutError:
|
||||
pass
|
||||
return '我网线被人拔了..QAQ', '', 999
|
||||
|
||||
|
||||
async def search_online_setu(url: str):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
for i in range(3):
|
||||
print(f'search_online_setu --> {i}')
|
||||
try:
|
||||
async with session.get(url, proxy=get_local_proxy(), timeout=7) as res:
|
||||
if res.status == 200:
|
||||
index = str(random.randint(1, 100000))
|
||||
async with aiofiles.open(IMAGE_PATH + 'temp/' + index + "_temp_setu.jpg", 'wb') as f:
|
||||
try:
|
||||
await f.write(await res.read())
|
||||
except TimeoutError:
|
||||
# return '\n这图没下载过来~（网太差？）', -1, False
|
||||
continue
|
||||
logger.info(f"下载 lolicon图片 {url} 成功， id：{index}")
|
||||
return image(f'{index}_temp_setu.jpg', 'temp'), index
|
||||
else:
|
||||
logger.warning(f"访问 lolicon图片 {url} 失败 status：{res.status}")
|
||||
# return '\n这图好难下载啊！QAQ', -1, False
|
||||
except TimeoutError:
|
||||
pass
|
||||
return '\n图片被小怪兽恰掉啦..!QAQ', -1
|
||||
|
||||
|
||||
def get_setu(index: str):
|
||||
length = len(os.listdir(IMAGE_PATH + path))
|
||||
if not index:
|
||||
index = random.randint(0, length - 1)
|
||||
if is_number(index):
|
||||
if int(index) > length or int(index) < 0:
|
||||
return f"超过当前上下限！({length - 1})", 999
|
||||
else:
|
||||
return f'id：{index}' + image(f'{index}.jpg', 'setu'), index
|
||||
return None, None
|
||||
|
||||
|
||||
def get_luoxiang(impression):
|
||||
probability = impression + 70
|
||||
if probability < random.randint(1, 101):
|
||||
return "我为什么要给你发这个？" + image(random.choice(os.listdir(IMAGE_PATH + "luoxiang/")), 'luoxiang') + \
|
||||
"\n(快向小真寻签到提升好感度吧！)"
|
||||
return None
|
||||
|
||||
|
||||
async def check_r18_and_keyword(msg: str, user_id) -> 'str, int, int':
|
||||
msg_list = msg.split(' ')
|
||||
num = 1
|
||||
r18 = 0
|
||||
keyword = ''
|
||||
if len(msg_list) == 1:
|
||||
if msg_list[0].strip().lower() in ['r', 'r18']:
|
||||
r18 = 1
|
||||
num = 10
|
||||
else:
|
||||
keyword = msg_list[0]
|
||||
elif len(msg_list) == 2:
|
||||
keyword = msg_list[1].strip()
|
||||
if msg_list[0].strip().lower() in ['r', 'r18']:
|
||||
r18 = 1
|
||||
num = 10
|
||||
else:
|
||||
keyword = msg[0]
|
||||
if r18 == 1:
|
||||
await UserCount.add_user(user_id)
|
||||
return keyword, r18, num
|
||||
|
||||
|
||||
# def delete_img(index):
|
||||
# if os.path.exists(IMAGE_PATH + path + f"{index}.jpg"):
|
||||
# img_hash = str(get_img_hash(IMAGE_PATH + f"setu/{index}.jpg"))
|
||||
# tp = list(setu_hash_dict.keys())[list(setu_hash_dict.values()).index(img_hash)]
|
||||
# logger.info(f"色图 {index}.jpg 与 {tp}.jpg 相似， 删除....")
|
||||
# os.remove(IMAGE_PATH + path + f"{index}.jpg")
|
||||
|
||||
|
||||
async def find_img_index(img_url, user_id):
|
||||
try:
|
||||
setu_hash_dict = json.load(open(TXT_PATH + 'setu_img_hash.json'))
|
||||
except (FileNotFoundError, ValueError):
|
||||
setu_hash_dict = {}
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.get(img_url, proxy=get_local_proxy(), timeout=5) as res:
|
||||
async with aiofiles.open(IMAGE_PATH + f"temp/{user_id}_find_setu_index.jpg", 'wb') as f:
|
||||
await f.write(await res.read())
|
||||
img_hash = str(get_img_hash(IMAGE_PATH + f"temp/{user_id}_find_setu_index.jpg"))
|
||||
try:
|
||||
tp = list(setu_hash_dict.keys())[list(setu_hash_dict.values()).index(img_hash)]
|
||||
return "id --> " + str(tp)
|
||||
except ValueError:
|
||||
return "该图不在色图库中！"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user