mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
80 lines
3.5 KiB
Python
80 lines
3.5 KiB
Python
|
from configs.path_config import IMAGE_PATH, TEMP_PATH
|
|||
|
|
import os
|
|||
|
|
from util.init_result import image
|
|||
|
|
from services.log import logger
|
|||
|
|
from nonebot import on_command
|
|||
|
|
from nonebot.rule import to_me
|
|||
|
|
from nonebot.typing import T_State
|
|||
|
|
from nonebot.adapters import Bot, Event
|
|||
|
|
from configs.config import IMAGE_DIR_LIST
|
|||
|
|
from util.utils import is_number, cn2py
|
|||
|
|
|
|||
|
|
__plugin_name__ = '删除图片'
|
|||
|
|
__plugin_usage__ = '删除图片帮助：\n\t' \
|
|||
|
|
'1.查看列表 --> 指令: 删除图片 列表/目录\n\t' \
|
|||
|
|
'2.删除图片 图库 id, 即在相应目录下删除图片\n\t\t示例: 删除图片 色图 1 '
|
|||
|
|
|
|||
|
|
|
|||
|
|
delete_img = on_command("删除图片", priority=5, rule=to_me(), block=True)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@delete_img.args_parser
|
|||
|
|
async def parse(bot: Bot, event: Event, state: T_State):
|
|||
|
|
if str(event.get_message()) in ['取消', '算了']:
|
|||
|
|
await delete_img.finish("已取消操作..", at_sender=True)
|
|||
|
|
if state["_current_key"] in ['path']:
|
|||
|
|
if str(event.get_message()) not in IMAGE_DIR_LIST:
|
|||
|
|
await delete_img.reject("此目录不正确，请重新输入目录！")
|
|||
|
|
state[state["_current_key"]] = str(event.get_message())
|
|||
|
|
if state["_current_key"] == 'id':
|
|||
|
|
if not is_number(str(event.get_message())):
|
|||
|
|
await delete_img.reject("id不正确！请重新输入数字...")
|
|||
|
|
state[state["_current_key"]] = str(event.get_message())
|
|||
|
|
|
|||
|
|
|
|||
|
|
@delete_img.handle()
|
|||
|
|
async def _(bot: Bot, event: Event, state: T_State):
|
|||
|
|
raw_arg = str(event.get_message()).strip()
|
|||
|
|
if raw_arg:
|
|||
|
|
args = raw_arg.split(" ")
|
|||
|
|
if args[0] in ['帮助']:
|
|||
|
|
await delete_img.finish(__plugin_usage__)
|
|||
|
|
if len(args) >= 2 and args[0] in IMAGE_DIR_LIST and is_number(args[1]):
|
|||
|
|
state['path'] = args[0]
|
|||
|
|
state['id'] = args[1]
|
|||
|
|
|
|||
|
|
|
|||
|
|
@delete_img.got("path", prompt="请输入要删除的目标图库？")
|
|||
|
|
@delete_img.got("id", prompt="请输入要删除的图片id？")
|
|||
|
|
async def arg_handle(bot: Bot, event: Event, state: T_State):
|
|||
|
|
path = cn2py(state['path'])
|
|||
|
|
img_id = state['id']
|
|||
|
|
path = IMAGE_PATH + path
|
|||
|
|
max_id = len(os.listdir(path)) - 1
|
|||
|
|
if int(img_id) > max_id or int(img_id) < 0:
|
|||
|
|
await delete_img.finish(f"Id超过上下限，上限：{max_id}", at_sender=True)
|
|||
|
|
try:
|
|||
|
|
if os.path.exists(IMAGE_PATH + TEMP_PATH + "delete.jpg"):
|
|||
|
|
os.remove(IMAGE_PATH + TEMP_PATH + "delete.jpg")
|
|||
|
|
logger.info("删除图片 delete.jpg 成功")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"删除图片 delete.jpg 失败 e{e}")
|
|||
|
|
try:
|
|||
|
|
os.rename(path + img_id + ".jpg", IMAGE_PATH + TEMP_PATH + "delete.jpg")
|
|||
|
|
logger.info(f"移动 {path}{img_id}.jpg 移动成功")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"{path}{img_id}.jpg --> 移动失败 e:{e}")
|
|||
|
|
if not os.path.exists(path + img_id + ".jpg"):
|
|||
|
|
try:
|
|||
|
|
if int(img_id) != max_id:
|
|||
|
|
os.rename(path + str(max_id) + ".jpg", path + img_id + ".jpg")
|
|||
|
|
except FileExistsError:
|
|||
|
|
logger.error(f"{path}{max_id}.jpg 替换 {path}{img_id}.jpg 失败 e:{e}")
|
|||
|
|
logger.info(f"{path}{max_id}.jpg 替换 {path}{img_id}.jpg 成功")
|
|||
|
|
logger.info(f"USER {event.user_id} GROUP {event.group_id if event.message_type != 'private' else 'private'}"
|
|||
|
|
f" -> id: {img_id} 删除成功")
|
|||
|
|
await delete_img.finish(f"id: {img_id} 删除成功" + image("delete.jpg", TEMP_PATH), at_sender=True)
|
|||
|
|
await delete_img.finish(f"id: {img_id} 删除失败！")
|
|||
|
|
|
|||
|
|
|