mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
296 lines
12 KiB
Python
Executable File
296 lines
12 KiB
Python
Executable File
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)
|