from nonebot import on_command from PIL import Image, ImageFilter from utils.message_builder import image from configs.path_config import TEMP_PATH, IMAGE_PATH from services.log import logger from nonebot.rule import to_me from nonebot.adapters.onebot.v11 import MessageEvent, GroupMessageEvent, Message from nonebot.typing import T_State from utils.utils import get_message_img, is_number from nonebot.params import CommandArg, Arg, ArgStr, Depends from utils.image_utils import BuildImage, pic2b64 from configs.config import NICKNAME from utils.http_utils import AsyncHttpx from typing import Union import cv2 import numpy as np __zx_plugin_name__ = "各种图片简易操作" __plugin_usage__ = """ usage： 简易的基础图片操作，输入 指定操作 或 序号 来进行选择 指令： 1.修改尺寸 [宽] [高] [图片] 2.等比压缩 [比例] [图片] 3.旋转图片 [角度] [图片] 4.水平翻转 [图片] 5.铅笔滤镜 [图片] 6.模糊效果 [图片] 7.锐化效果 [图片] 8.高斯模糊 [图片] 9.边缘检测 [图片] 10.底色替换 [红/蓝] [红/蓝/白/绿/黄] [图片] 示例：图片修改尺寸 100 200 [图片] 示例：图片 2 0.3 [图片] """.strip() __plugin_des__ = "10种快捷的图片简易操作" __plugin_cmd__ = [ "改图 修改尺寸 [宽] [高] [图片]", "改图 等比压缩 [比例] [图片]", "改图 旋转图片 [角度] [图片]", "改图 水平翻转 [图片]", "改图 铅笔滤镜 [图片]", "改图 模糊效果 [图片]", "改图 锐化效果 [图片]", "改图 高斯模糊 [图片]", "改图 边缘检测 [图片]", "改图 底色替换 [红/蓝] [红/蓝/白/绿/黄] [图片]", ] __plugin_type__ = ("一些工具", 1) __plugin_version__ = 0.1 __plugin_author__ = "HibiKier" __plugin_settings__ = { "level": 5, "default_status": True, "limit_superuser": False, "cmd": ["修改图片", "改图", "操作图片"], } 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 = BuildImage(960, 700, font_size=24) update_img_help.text((10, 10), __plugin_usage__) update_img_help.save(IMAGE_PATH / "update_img_help.png") def parse_key(key: str): async def _key_parser( state: T_State, inp: Union[Message, str] = Arg(key) ): if key != "img_list" and isinstance(inp, Message): inp = inp.extract_plain_text().strip() if inp in ["取消", "算了"]: await update_img.finish("已取消操作..") if key == "method": if inp not in method_oper: await update_img.reject_arg("method", f"操作不正确，请重新输入！{method_str}") elif key == "x": method = state["method"] if method in ["1", "修改尺寸"]: if not is_number(inp) or int(inp) < 1: await update_img.reject_arg("x", "宽度不正确！请重新输入数字...") elif method in ["2", "等比压缩", "3", "旋转图片"]: if not is_number(inp): await update_img.reject_arg("x", "比率不正确！请重新输入数字...") elif method in ["10", "底色替换"]: if inp not in ["红色", "蓝色", "红", "蓝"]: await update_img.reject_arg("x", "请输入支持的被替换的底色：\n红色 蓝色") elif key == "y": method = state["method"] if method in ["1", "修改尺寸"]: if not is_number(inp) or int(inp) < 1: await update_img.reject_arg("y", "长度不正确！请重新输入数字...") elif method in ["10", "底色替换"]: if inp not in [ "红色", "白色", "蓝色", "绿色", "黄色", "红", "白", "蓝", "绿", "黄", ]: await update_img.reject_arg("y", "请输入支持的替换的底色：\n红色 蓝色 白色 绿色") elif key == "img_list": if not get_message_img(inp): await update_img.reject_arg("img_list", "没图？没图？没图？来图速来！") state[key] = inp return _key_parser @update_img.handle() async def _(event: MessageEvent, state: T_State, arg: Message = CommandArg()): if str(event.get_message()) in ["帮助"]: await update_img.finish(image("update_img_help.png")) raw_arg = arg.extract_plain_text().strip() img_list = get_message_img(event.json()) if raw_arg: args = raw_arg.split("[")[0].split() 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["img_list"] = event.message @update_img.got("method", prompt=f"要使用图片的什么操作呢？{method_str}", parameterless=[Depends(parse_key("method"))]) @update_img.got("x", prompt="[宽度？ 比率？ 旋转角度？ 底色？]", parameterless=[Depends(parse_key("x"))]) @update_img.got("y", prompt="[长度？ 0 0 底色？]", parameterless=[Depends(parse_key("y"))]) @update_img.got("img_list", prompt="图呢图呢图呢图呢？GKD！", parameterless=[Depends(parse_key("img_list"))]) async def _( event: MessageEvent, state: T_State, method: str = ArgStr("method"), x: str = ArgStr("x"), y: str = ArgStr("y"), img_list: Message = Arg("img_list"), ): x = x or "" y = y or "" img_list = get_message_img(img_list) if is_number(x): x = float(x) if is_number(y): y = int(y) index = 0 result = "" for img_url in img_list: if await AsyncHttpx.download_file( img_url, TEMP_PATH / f"{event.user_id}_{index}_update.png" ): index += 1 else: await update_img.finish("获取图片超时了...", at_sender=True) if index == 0: return if method in ["修改尺寸", "1"]: for i in range(index): img = Image.open(TEMP_PATH / f"{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(TEMP_PATH / f"{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(f"{NICKNAME}不支持图片压缩后宽或高大于8000的存在！！") if method in ["旋转图片", "3"]: for i in range(index): img = Image.open(TEMP_PATH / f"{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(TEMP_PATH / f"{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(TEMP_PATH / f"{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(TEMP_PATH / f"{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(TEMP_PATH / f"{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(TEMP_PATH / f"{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(TEMP_PATH / f"{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(TEMP_PATH / f"{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(TEMP_PATH / f"{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 isinstance(event, GroupMessageEvent) else 'private'}) 使用{method}" ) await update_img.finish(result, at_sender=True)