from nonebot import on_command from PIL import Image, ImageFilter from util.init_result import image from configs.path_config import IMAGE_PATH from services.log import logger from nonebot.rule import to_me from nonebot.adapters.cqhttp import Bot, MessageEvent from nonebot.typing import T_State from util.utils import get_message_imgs import aiofiles import aiohttp from util.utils import is_number, get_message_text from util.img_utils import CreateImg, pic2b64 import cv2 import numpy as np __plugin_name__ = '图片' __plugin_usage__ = '''图片用法(目前): 可以输入 指定操作 或 序号 来进行选择 1.修改尺寸 宽(x) 高(y) 图片(在文字后跟图片即可) 示例：小真寻修改图片 修改尺寸 200 300 图片(在文字后跟图片即可) 2.等比压缩 比例(x) 图片(在文字后跟图片即可) 示例：小真寻修改图片 等比压缩 2 图片(在文字后跟图片即可) 3.旋转图片 角度(x) 图片(在文字后跟图片即可) 示例：小真寻修改图片 旋转图片 45 图片(在文字后跟图片即可) 4.水平翻转 示例：小真寻修改图片 水平翻转 图片 5.铅笔滤镜 示例：小真寻修改图片 铅笔滤镜 图片 6.模糊效果 示例：小真寻修改图片 模糊效果 图片 7.锐化效果 示例：小真寻修改图片 锐化效果 图片 8.高斯模糊 示例：小真寻修改图片 高斯模糊 图片 9.边缘检测 示例：小真寻修改图片 边缘检测 图片 10.底色替换 颜色 颜色 图片 （不专业）支持：（红，蓝）->（红，蓝，白，绿，黄） 示例：小真寻修改图片 底色替换 蓝色 红色 图片''' # IMAGE_LOCAL = IMAGE_PATH + "temp/{}_update.png" method_flag = '' update_img = on_command("修改图片", aliases={'图片', '操作图片', '改图'}, priority=5, rule=to_me(), block=True) method_list = ['修改尺寸', '等比压缩', '旋转图片', '水平翻转', '铅笔滤镜', '模糊效果', '锐化效果', '高斯模糊', '边缘检测', '底色替换'] method_str = '' method_oper = [] for i in range(len(method_list)): method_str += f'\n{i + 1}.{method_list[i]}' method_oper.append(method_list[i]) method_oper.append(str(i + 1)) update_img_help = CreateImg(960, 700, font_size=24) update_img_help.text((10, 10), __plugin_usage__) update_img_help.save(IMAGE_PATH + 'update_img_help.png') @update_img.args_parser async def _(bot: Bot, event: MessageEvent, state: T_State): global method_flag if str(event.get_message()) in ['取消', '算了']: await update_img.finish("已取消操作..", at_sender=True) if state["_current_key"] in ['method']: if str(event.get_message()) not in method_oper: await update_img.reject(f"操作不正确，请重新输入！{method_str}") state[state["_current_key"]] = str(event.get_message()) method_flag = str(event.get_message()) if method_flag in ['1', '修改尺寸']: if state["_current_key"] == 'x': if not is_number(str(event.get_message())): await update_img.reject("宽度不正确！请重新输入数字...") state[state["_current_key"]] = str(event.get_message()) if state["_current_key"] == 'y': if not is_number(str(event.get_message())): await update_img.reject("长度不正确！请重新输入数字...") state[state["_current_key"]] = str(event.get_message()) elif method_flag in ['2', '等比压缩', '3', '旋转图片']: if state["_current_key"] == 'x': if not is_number(str(event.get_message())): await update_img.reject("比率不正确！请重新输入数字...") state[state["_current_key"]] = str(event.get_message()) state['y'] = '' elif method_flag in ['4', '水平翻转', '5', '铅笔滤镜', '6', '模糊效果', '7', '锐化效果', '8', '高斯模糊', '9', '边缘检测']: state['x'] = '' state['y'] = '' elif method_flag in ['10', '底色替换']: if state["_current_key"] == 'x': if str(event.get_message()) not in ['红色', '蓝色', '红', '蓝']: await update_img.reject('请输入支持的被替换的底色：\n红色 蓝色') state['x'] = str(event.get_message()) if state["_current_key"] == 'y': if str(event.get_message()) not in ['红色', '白色', '蓝色', '绿色', '黄色', '红', '白', '蓝', '绿', '黄']: await update_img.reject('请输入支持的替换的底色：\n红色 蓝色 白色 绿色') state['y'] = str(event.get_message()) if state["_current_key"] == 'imgs': if not get_message_imgs(event.json()): await update_img.reject("没图？没图？没图？来图速来！") state[state["_current_key"]] = get_message_imgs(event.json()) @update_img.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): if str(event.get_message()) in ['帮助']: await update_img.finish(image('update_img_help.png')) raw_arg = get_message_text(event.json()) img_list = get_message_imgs(event.json()) if raw_arg: args = raw_arg.split("[")[0].split(" ") print(args) state['method'] = args[0] if len(args) == 2: if args[0] in ["等比压缩", '旋转图片']: if is_number(args[1]): state['x'] = args[1] state['y'] = '' elif len(args) > 2: if args[0] in ["修改尺寸"]: if is_number(args[1]): state['x'] = args[1] if is_number(args[2]): state['y'] = args[2] if args[0] in ['底色替换']: if args[1] in ['红色', '蓝色', '蓝', '红']: state['x'] = args[1] if args[2] in ['红色', '白色', '蓝色', '绿色', '黄色', '红', '白', '蓝', '绿', '黄']: state['y'] = args[2] if args[0] in ['水平翻转', '铅笔滤镜', '模糊效果', '锐化效果', '高斯模糊', '边缘检测']: state['x'] = '' state['y'] = '' if img_list: state['imgs'] = img_list @update_img.got("method", prompt=f"要使用图片的什么操作呢？{method_str}") @update_img.got("x", prompt="[宽度？ 比率？ 旋转角度？ 底色？]") @update_img.got("y", prompt="[长度？ 0 0 底色？]") @update_img.got("imgs", prompt="图呢图呢图呢图呢？GKD！") async def _(bot: Bot, event: MessageEvent, state: T_State): method = state['method'] x = state['x'] if state['x'] else '' y = state['y'] if state['y'] else '' img_list = state['imgs'] if is_number(x): x = float(x) if is_number(y): y = int(y) index = 0 result = '' async with aiohttp.ClientSession() as session: for img_url in img_list: async with session.get(img_url, timeout=7) as response: if response.status == 200: async with aiofiles.open(IMAGE_PATH + f'temp/{event.user_id}_{index}_update.png', 'wb') as f: await f.write(await response.read()) index += 1 else: logger.warning(f"USER {event.user_id} GROUP " f"{event.group_id if event.message_type != 'private' else 'private'} " f"使用 {method} 时下载图片超时") await update_img.finish("获取图片超时了...", at_sender=True) if index == 0: return if method in ['修改尺寸', '1']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png') img = img.convert('RGB') img = img.resize((int(x), int(y)), Image.ANTIALIAS) result += image(b64=pic2b64(img)) await update_img.finish(result, at_sender=True) if method in ['等比压缩', '2']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png') width, height = img.size img = img.convert('RGB') if width * x < 8000 and height * x < 8000: img = img.resize((int(x * width), int(x * height))) result += image(b64=pic2b64(img)) else: await update_img.finish('小真寻不支持图片压缩后宽或高大于8000的存在！！') if method in ['旋转图片', '3']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png') img = img.rotate(x) result += image(b64=pic2b64(img)) if method in ['水平翻转', '4']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png') img = img.transpose(Image.FLIP_LEFT_RIGHT) result += image(b64=pic2b64(img)) if method in ['铅笔滤镜', '5']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png').filter(ImageFilter.CONTOUR) result += image(b64=pic2b64(img)) if method in ['模糊效果', '6']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png').filter(ImageFilter.BLUR) result += image(b64=pic2b64(img)) if method in ['锐化效果', '7']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png').filter(ImageFilter.EDGE_ENHANCE) result += image(b64=pic2b64(img)) if method in ['高斯模糊', '8']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png').filter(ImageFilter.GaussianBlur) result += image(b64=pic2b64(img)) if method in ['边缘检测', '9']: for i in range(index): img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png').filter(ImageFilter.FIND_EDGES) result += image(b64=pic2b64(img)) if method in ['底色替换', '10']: if x in ['蓝色', '蓝']: lower = np.array([90, 70, 70]) upper = np.array([110, 255, 255]) if x in ['红色', '红']: lower = np.array([0, 135, 135]) upper = np.array([180, 245, 230]) if y in ['蓝色', '蓝']: color = (255, 0, 0) if y in ['红色', '红']: color = (0, 0, 255) if y in ['白色', '白']: color = (255, 255, 255) if y in ['绿色', '绿']: color = (0, 255, 0) if y in ['黄色', '黄']: color = (0, 255, 255) for k in range(index): img = cv2.imread(IMAGE_PATH + f'temp/{event.user_id}_{k}_update.png') img = cv2.resize(img, None, fx=0.3, fy=0.3) rows, cols, channels = img.shape hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) mask = cv2.inRange(hsv, lower, upper) # erode = cv2.erode(mask, None, iterations=1) dilate = cv2.dilate(mask, None, iterations=1) for i in range(rows): for j in range(cols): if dilate[i, j] == 255: img[i, j] = color cv2.imwrite(IMAGE_PATH + f'temp/{event.user_id}_{k}_ok_update.png', img) for i in range(index): result += image(f'{event.user_id}_{i}_ok_update.png', 'temp') if is_number(method): method = method_list[int(method) - 1] logger.info( f"(USER {event.user_id}, GROUP" f" {event.group_id if event.message_type != 'private' else 'private'}) 使用{method}") await update_img.finish(result, at_sender=True)