mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from nonebot import on_command, on_keyword
|
||
from configs.path_config import IMAGE_PATH
|
||
from util.init_result import image
|
||
import os
|
||
import random
|
||
from util.utils import is_number
|
||
from services.log import logger
|
||
from nonebot.typing import T_State
|
||
from nonebot.adapters import Bot, Event
|
||
from util.utils import FreqLimiter, cn2py
|
||
from models.group_remind import GroupRemind
|
||
from configs.config import IMAGE_DIR_LIST
|
||
|
||
|
||
__plugin_name__ = '壁纸/萝莉/美图'
|
||
__plugin_usage__ = '用法： 发送"壁纸/萝莉/美图", 回复图片，后添加id获得指定图片'
|
||
|
||
_flmt = FreqLimiter(1)
|
||
|
||
IMAGE_DIR_LIST.remove('色图')
|
||
cmd = set(IMAGE_DIR_LIST)
|
||
|
||
# print(cmd)
|
||
|
||
send_img = on_command("img", aliases=cmd, priority=5, block=True)
|
||
|
||
|
||
@send_img.handle()
|
||
async def _(bot: Bot, event: Event, state: T_State):
|
||
img_id = str(event.get_message())
|
||
if img_id in ['帮助']:
|
||
await send_img.finish(__plugin_usage__)
|
||
path = cn2py(state["_prefix"]["raw_command"]) + '/'
|
||
if not os.path.exists(IMAGE_PATH + path):
|
||
logger.warning(f'未找到 {path} 文件夹，调用取消!')
|
||
return
|
||
length = len(os.listdir(IMAGE_PATH + path)) - 1
|
||
index = img_id if img_id else str(random.randint(0, length))
|
||
if not is_number(index):
|
||
await send_img.finish("id错误！")
|
||
if int(index) > length or int(index) < 0:
|
||
await send_img.finish(f"超过当前上下限！({length - 1})")
|
||
result = image(f'{index}.jpg', path)
|
||
if result:
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'}) 发送{path}:" + result)
|
||
await send_img.finish(f"id：{index}" + result)
|
||
else:
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'}) 发送 {path} 失败")
|
||
await send_img.finish(f"不想给你看Ov|")
|
||
|
||
|
||
pa = on_keyword({"爬"}, priority=1, block=True)
|
||
|
||
|
||
@pa.handle()
|
||
async def _(bot: Bot, event: Event, state: T_State):
|
||
if await GroupRemind.get_status(event.group_id, 'pa'):
|
||
try:
|
||
if str(event.get_message()[:2]) in ['开启', '关闭']:
|
||
return
|
||
except:
|
||
return
|
||
if not _flmt.check(event.user_id):
|
||
return
|
||
_flmt.start_cd(event.user_id)
|
||
await pa.finish(image(random.choice(os.listdir(IMAGE_PATH + "pa")), 'pa'))
|