mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-05 03:39:59 +08:00
refactor code
This commit is contained in:
hibiki
+144
-93
@@ -1,22 +1,22 @@
|
||||
from nonebot import on_command
|
||||
from PIL import Image, ImageFilter
|
||||
from utils.init_result import image
|
||||
from utils.message_builder 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.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent
|
||||
from nonebot.typing import T_State
|
||||
from utils.utils import get_message_imgs
|
||||
import aiofiles
|
||||
import aiohttp
|
||||
from utils.utils import is_number, get_message_text
|
||||
from utils.img_utils import CreateImg, pic2b64
|
||||
from utils.image_utils import CreateImg, pic2b64
|
||||
import cv2
|
||||
import numpy as np
|
||||
|
||||
|
||||
__plugin_name__ = '图片'
|
||||
__plugin_usage__ = '''图片用法(目前):
|
||||
__plugin_name__ = "图片"
|
||||
__plugin_usage__ = """图片用法(目前):
|
||||
可以输入 指定操作 或 序号 来进行选择
|
||||
1.修改尺寸 宽(x) 高(y) 图片(在文字后跟图片即可)
|
||||
示例：小真寻修改图片 修改尺寸 200 300 图片(在文字后跟图片即可)
|
||||
@@ -37,64 +37,101 @@ __plugin_usage__ = '''图片用法(目前):
|
||||
9.边缘检测
|
||||
示例：小真寻修改图片 边缘检测 图片
|
||||
10.底色替换 颜色 颜色 图片 （不专业）支持：（红，蓝）->（红，蓝，白，绿，黄）
|
||||
示例：小真寻修改图片 底色替换 蓝色 红色 图片'''
|
||||
示例：小真寻修改图片 底色替换 蓝色 红色 图片"""
|
||||
|
||||
# IMAGE_LOCAL = IMAGE_PATH + "temp/{}_update.png"
|
||||
method_flag = ''
|
||||
method_flag = ""
|
||||
|
||||
update_img = on_command("修改图片", aliases={'图片', '操作图片', '改图'}, priority=5, rule=to_me(), block=True)
|
||||
update_img = on_command(
|
||||
"修改图片", aliases={"图片", "操作图片", "改图"}, priority=5, rule=to_me(), block=True
|
||||
)
|
||||
|
||||
method_list = ['修改尺寸', '等比压缩', '旋转图片', '水平翻转', '铅笔滤镜', '模糊效果', '锐化效果', '高斯模糊', '边缘检测', '底色替换']
|
||||
method_str = ''
|
||||
method_list = [
|
||||
"修改尺寸",
|
||||
"等比压缩",
|
||||
"旋转图片",
|
||||
"水平翻转",
|
||||
"铅笔滤镜",
|
||||
"模糊效果",
|
||||
"锐化效果",
|
||||
"高斯模糊",
|
||||
"边缘检测",
|
||||
"底色替换",
|
||||
]
|
||||
method_str = ""
|
||||
method_oper = []
|
||||
for i in range(len(method_list)):
|
||||
method_str += f'\n{i + 1}.{method_list[i]}'
|
||||
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_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 ['取消', '算了']:
|
||||
if str(event.get_message()) in ["取消", "算了"]:
|
||||
await update_img.finish("已取消操作..", at_sender=True)
|
||||
if state["_current_key"] in ['method']:
|
||||
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 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 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':
|
||||
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':
|
||||
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())
|
||||
@@ -102,35 +139,35 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
|
||||
@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'))
|
||||
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]
|
||||
state["method"] = args[0]
|
||||
if len(args) == 2:
|
||||
if args[0] in ["等比压缩", '旋转图片']:
|
||||
if args[0] in ["等比压缩", "旋转图片"]:
|
||||
if is_number(args[1]):
|
||||
state['x'] = args[1]
|
||||
state['y'] = ''
|
||||
state["x"] = args[1]
|
||||
state["y"] = ""
|
||||
elif len(args) > 2:
|
||||
if args[0] in ["修改尺寸"]:
|
||||
if is_number(args[1]):
|
||||
state['x'] = 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'] = ''
|
||||
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
|
||||
state["imgs"] = img_list
|
||||
|
||||
|
||||
@update_img.got("method", prompt=f"要使用图片的什么操作呢？{method_str}")
|
||||
@@ -138,96 +175,110 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
@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']
|
||||
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 = ''
|
||||
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:
|
||||
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} 时下载图片超时")
|
||||
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']:
|
||||
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 = 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']:
|
||||
if method in ["等比压缩", "2"]:
|
||||
for i in range(index):
|
||||
img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png')
|
||||
img = Image.open(IMAGE_PATH + f"temp/{event.user_id}_{i}_update.png")
|
||||
width, height = img.size
|
||||
img = img.convert('RGB')
|
||||
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']:
|
||||
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 = 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']:
|
||||
if method in ["水平翻转", "4"]:
|
||||
for i in range(index):
|
||||
img = Image.open(IMAGE_PATH + f'temp/{event.user_id}_{i}_update.png')
|
||||
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']:
|
||||
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)
|
||||
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']:
|
||||
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)
|
||||
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']:
|
||||
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)
|
||||
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']:
|
||||
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)
|
||||
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']:
|
||||
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)
|
||||
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 ['蓝色', '蓝']:
|
||||
if method in ["底色替换", "10"]:
|
||||
if x in ["蓝色", "蓝"]:
|
||||
lower = np.array([90, 70, 70])
|
||||
upper = np.array([110, 255, 255])
|
||||
if x in ['红色', '红']:
|
||||
if x in ["红色", "红"]:
|
||||
lower = np.array([0, 135, 135])
|
||||
upper = np.array([180, 245, 230])
|
||||
if y in ['蓝色', '蓝']:
|
||||
if y in ["蓝色", "蓝"]:
|
||||
color = (255, 0, 0)
|
||||
if y in ['红色', '红']:
|
||||
if y in ["红色", "红"]:
|
||||
color = (0, 0, 255)
|
||||
if y in ['白色', '白']:
|
||||
if y in ["白色", "白"]:
|
||||
color = (255, 255, 255)
|
||||
if y in ['绿色', '绿']:
|
||||
if y in ["绿色", "绿"]:
|
||||
color = (0, 255, 0)
|
||||
if y in ['黄色', '黄']:
|
||||
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.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)
|
||||
@@ -238,13 +289,13 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
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)
|
||||
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')
|
||||
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}")
|
||||
f" {event.group_id if isinstance(event, GroupMessageEvent) else 'private'}) 使用{method}"
|
||||
)
|
||||
await update_img.finish(result, at_sender=True)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user