mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
332 lines
14 KiB
Python
332 lines
14 KiB
Python
import random
|
||
from nonebot import on_command, on_regex
|
||
from nonebot.permission import SUPERUSER
|
||
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, delete_img, add_img
|
||
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 MAX_SETU_R_COUNT
|
||
from configs.path_config import TXT_PATH
|
||
try:
|
||
import ujson as json
|
||
except ModuleNotFoundError:
|
||
import json
|
||
|
||
__plugin_name__ = '色图'
|
||
__plugin_usage__ = f'''示例：
|
||
1. 色图 （随机本地色图）
|
||
2. 色图r （随机在线十张r18涩图）
|
||
3. 色图 666 （本地色图id）
|
||
4. 色图 xx （在线搜索xx色图）
|
||
5. 色图r xx （搜索十张xx的r18涩图，注意空格）（仅私聊，每日限制5次）
|
||
6. 来n张涩图 （本地涩图连发）（1<=n<=9）
|
||
7. 来n张xx的涩图 （在线搜索xx涩图）（较慢，看网速）
|
||
注：【色图r每日提供{MAX_SETU_R_COUNT}次
|
||
本地涩图没有r18！
|
||
联网搜索会较慢！
|
||
如果图片数量与数字不符，
|
||
原因1：网络不好，网线被拔QAQ
|
||
原因2：搜索到的总数小于数字
|
||
原因3：图太色或者小错误了】'''
|
||
|
||
url = "https://api.lolicon.app/setu/"
|
||
_flmt = FreqLimiter(5)
|
||
_ulmt = UserExistLimiter()
|
||
path = "_setu/"
|
||
|
||
|
||
setu = on_command("色图", aliases={"涩图", "不够色", "来一发", "再来点"}, priority=5, block=True)
|
||
setu_reg = on_regex('(.*)[份|发|张|个|次|点](.*)[瑟|色|涩]图', priority=5, block=True)
|
||
find_setu = on_command("查色图", priority=5, block=True)
|
||
delete_setu = on_command('删除色图', aliases={'删除涩图'}, priority=5, block=True, permission=SUPERUSER)
|
||
upload_setu = on_command('上传色图', aliases={'上传涩图'}, priority=5, block=True, permission=SUPERUSER)
|
||
|
||
|
||
@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)
|
||
try:
|
||
setu_data = json.load(open(f'{TXT_PATH}/setu_data.json', 'r', encoding='utf8'))
|
||
except Exception:
|
||
setu_data = {}
|
||
setu_img, index = get_setu(img_id, setu_data)
|
||
if setu_img:
|
||
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 list(bot.config.nickname)[0].find(msg) != -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:
|
||
_ulmt.set_False(event.user_id)
|
||
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)
|
||
try:
|
||
setu_data = json.load(open(f'{TXT_PATH}/setu_data.json', 'r', encoding='utf8'))
|
||
except Exception:
|
||
setu_data = {}
|
||
setu_img, index = get_setu(img_id, setu_data)
|
||
if setu_img:
|
||
await setu.send(setu_img)
|
||
logger.info(
|
||
f"USER {event.user_id} GROUP private 发送色图 {index}.jpg 成功")
|
||
else:
|
||
msg = img_id
|
||
if list(bot.config.nickname)[0].find(msg) != -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_SETU_R_COUNT):
|
||
_ulmt.set_False(event.user_id)
|
||
await setu.finish('要节制啊，请明天再来...\n【每日提供5次】', at_sender=True)
|
||
else:
|
||
await UserCount.add_count(event.user_id, 'setu_r18', count=1)
|
||
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:
|
||
try:
|
||
setu_data = json.load(open(f'{TXT_PATH}/setu_data.json', 'r', encoding='utf8'))
|
||
except Exception:
|
||
setu_data = {}
|
||
if not keyword:
|
||
for _ in range(num):
|
||
try:
|
||
img, index = get_setu('', setu_data)
|
||
if not img:
|
||
break
|
||
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} 色图成功")
|
||
_ulmt.set_False(event.user_id)
|
||
else:
|
||
_ulmt.set_False(event.user_id)
|
||
return
|
||
if list(bot.config.nickname)[0].find(keyword) != -1:
|
||
await setu.finish('咳咳咳，虽然我很可爱，但是我木有自己的色图~~~有的话记得发我一份呀')
|
||
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)
|
||
|
||
|
||
@delete_setu.handle()
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
_id = get_message_text(event.json())
|
||
if _id:
|
||
flag, text = delete_img(int(_id))
|
||
if flag:
|
||
await delete_setu.finish(f'删除色图 id：{_id} 成功', at_sender=True)
|
||
else:
|
||
await delete_setu.finish(f'删除色图 id：{_id} 失败，{text}', at_sender=True)
|
||
|
||
|
||
@upload_setu.handle()
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
imgs = get_message_imgs(event.json())
|
||
if imgs:
|
||
lens, add_count = await add_img(imgs)
|
||
if add_count == 0:
|
||
await upload_setu.finish('上传的涩图已在色图库中，未成功上传，可以通过 查色图 来查看色图在图库中的id', at_sender=True)
|
||
id_s = ''
|
||
for i in range(add_count, 0, -1):
|
||
id_s += f'{lens - add_count} '
|
||
await upload_setu.finish(f"这次一共为 色图 库 添加了 {add_count} 张图片\n"
|
||
f"依次的Id为：{id_s[:-1]}\n"
|
||
f"小真寻感谢您对图库的扩充!WW", at_sender=True)
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'})"
|
||
f" 上传色图ID：{id_s[:-1]}")
|
||
|
||
|