mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
657 lines
18 KiB
Python
657 lines
18 KiB
Python
|
import base64
|
|||
|
|
import math
|
|||
|
|
from io import BytesIO
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import List, Literal, Tuple, TypeAlias, overload
|
|||
|
|
|
|||
|
|
from nonebot.utils import run_sync
|
|||
|
|
from PIL import Image, ImageDraw, ImageFilter, ImageFont
|
|||
|
|
from PIL.Image import Image as tImage
|
|||
|
|
from PIL.ImageFont import FreeTypeFont
|
|||
|
|
from typing_extensions import Self
|
|||
|
|
|
|||
|
|
from zhenxun.configs.path_config import FONT_PATH
|
|||
|
|
|
|||
|
|
ModeType = Literal[
|
|||
|
|
"1", "CMYK", "F", "HSV", "I", "L", "LAB", "P", "RGB", "RGBA", "RGBX", "YCbCr"
|
|||
|
|
]
|
|||
|
|
"""图片类型"""
|
|||
|
|
|
|||
|
|
ColorAlias: TypeAlias = str | Tuple[int, int, int] | Tuple[int, int, int, int] | None
|
|||
|
|
|
|||
|
|
CenterType = Literal["center", "height", "width"]
|
|||
|
|
"""
|
|||
|
|
粘贴居中
|
|||
|
|
|
|||
|
|
center: 水平垂直居中
|
|||
|
|
|
|||
|
|
height: 垂直居中
|
|||
|
|
|
|||
|
|
width: 水平居中
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class BuildImage:
|
|||
|
|
"""
|
|||
|
|
快捷生成图片与操作图片的工具类
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def __init__(
|
|||
|
|
self,
|
|||
|
|
width: int = 0,
|
|||
|
|
height: int = 0,
|
|||
|
|
color: ColorAlias = None,
|
|||
|
|
mode: ModeType = "RGBA",
|
|||
|
|
font: str | Path | FreeTypeFont = "HYWenHei-85W.ttf",
|
|||
|
|
font_size: int = 20,
|
|||
|
|
background: str | BytesIO | Path | None = None,
|
|||
|
|
) -> None:
|
|||
|
|
self.width = width
|
|||
|
|
self.height = height
|
|||
|
|
self.color = color
|
|||
|
|
self.font = (
|
|||
|
|
self.load_font(font, font_size)
|
|||
|
|
if not isinstance(font, FreeTypeFont)
|
|||
|
|
else font
|
|||
|
|
)
|
|||
|
|
if background:
|
|||
|
|
self.markImg = Image.open(background)
|
|||
|
|
if width and height:
|
|||
|
|
self.markImg = self.markImg.resize((width, height), Image.LANCZOS)
|
|||
|
|
else:
|
|||
|
|
if not width or not height:
|
|||
|
|
raise ValueError("长度和宽度不能为空...")
|
|||
|
|
self.markImg = Image.new(mode, (width, height), color) # type: ignore
|
|||
|
|
self.draw = ImageDraw.Draw(self.markImg)
|
|||
|
|
|
|||
|
|
@property
|
|||
|
|
def size(self) -> Tuple[int, int]:
|
|||
|
|
return self.markImg.size
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
async def build_text_image(
|
|||
|
|
cls,
|
|||
|
|
text: str,
|
|||
|
|
font: str | Path = "HYWenHei-85W.ttf",
|
|||
|
|
size: int = 10,
|
|||
|
|
font_color: str | Tuple[int, int, int] = (0, 0, 0),
|
|||
|
|
color: ColorAlias = None,
|
|||
|
|
padding: int | Tuple[int, int, int, int] | None = None,
|
|||
|
|
) -> Self:
|
|||
|
|
"""构建文本图片
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
text: 文本
|
|||
|
|
font: 字体路径
|
|||
|
|
size: 字体大小
|
|||
|
|
font_color: 字体颜色.
|
|||
|
|
color: 背景颜色
|
|||
|
|
padding: 外边距
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
Self: Self
|
|||
|
|
"""
|
|||
|
|
_font = cls.load_font(font, size)
|
|||
|
|
width, height = cls.get_text_size(text, _font)
|
|||
|
|
if type(padding) == int:
|
|||
|
|
width += padding * 2
|
|||
|
|
height += padding * 2
|
|||
|
|
elif type(padding) == tuple:
|
|||
|
|
width += padding[1] + padding[3]
|
|||
|
|
height += padding[0] + padding[2]
|
|||
|
|
markImg = cls(width, height, color)
|
|||
|
|
await markImg.text(
|
|||
|
|
(0, 0), text, fill=font_color, font=_font, center_type="center"
|
|||
|
|
)
|
|||
|
|
return markImg
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
async def auto_paste(
|
|||
|
|
cls,
|
|||
|
|
img_list: list[Self | tImage],
|
|||
|
|
row: int,
|
|||
|
|
space: int = 10,
|
|||
|
|
padding: int = 50,
|
|||
|
|
color: ColorAlias = (255, 255, 255, 0),
|
|||
|
|
background: str | BytesIO | Path | None = None,
|
|||
|
|
) -> Self | None:
|
|||
|
|
"""自动贴图
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
img_list: 图片列表
|
|||
|
|
row: 一行图片的数量
|
|||
|
|
space: 图片之间的间距.
|
|||
|
|
padding: 外边距.
|
|||
|
|
color: 图片背景颜色.
|
|||
|
|
background: 图片背景图片.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
Self: Self
|
|||
|
|
"""
|
|||
|
|
if not img_list:
|
|||
|
|
return None
|
|||
|
|
width, height = img_list[0].size
|
|||
|
|
background_width = width * row + space * (row - 1) + padding * 2
|
|||
|
|
column = math.ceil(len(img_list) / row)
|
|||
|
|
background_height = height * column + space * (column - 1) + padding * 2
|
|||
|
|
background_image = cls(
|
|||
|
|
background_width, background_height, color=color, background=background
|
|||
|
|
)
|
|||
|
|
_cur_width, _cur_height = padding, padding
|
|||
|
|
for img in img_list:
|
|||
|
|
await background_image.paste(img, (_cur_width, _cur_height))
|
|||
|
|
_cur_width += space + img.width
|
|||
|
|
_cur_height += space + img.height
|
|||
|
|
if _cur_width + padding >= background_image.width:
|
|||
|
|
_cur_width = padding
|
|||
|
|
return background_image
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def load_font(cls, font: str | Path, font_size: int) -> FreeTypeFont:
|
|||
|
|
"""
|
|||
|
|
加载字体
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
font: 字体名称
|
|||
|
|
font_size: 字体大小
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
FreeTypeFont: 字体
|
|||
|
|
"""
|
|||
|
|
path = FONT_PATH / font if type(font) == str else font
|
|||
|
|
return ImageFont.truetype(str(path), font_size)
|
|||
|
|
|
|||
|
|
@overload
|
|||
|
|
@classmethod
|
|||
|
|
def get_text_size(
|
|||
|
|
cls, text: str, font: FreeTypeFont | None = None
|
|||
|
|
) -> Tuple[int, int]:
|
|||
|
|
...
|
|||
|
|
|
|||
|
|
@overload
|
|||
|
|
@classmethod
|
|||
|
|
def get_text_size(
|
|||
|
|
cls, text: str, font: str | None = None, font_size: int = 10
|
|||
|
|
) -> Tuple[int, int]:
|
|||
|
|
...
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def get_text_size(
|
|||
|
|
cls, text: str, font: str | FreeTypeFont | None = None, font_size: int = 10
|
|||
|
|
) -> Tuple[int, int]:
|
|||
|
|
"""获取该字体下文本需要的长宽
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
text: 文本内容
|
|||
|
|
font: 字体名称或FreeTypeFont
|
|||
|
|
font_size: 字体大小
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
Tuple[int, int]: 长宽
|
|||
|
|
"""
|
|||
|
|
_font = font
|
|||
|
|
if font and type(font) == str:
|
|||
|
|
_font = cls.load_font(font, font_size)
|
|||
|
|
return _font.getsize(text) # type: ignore
|
|||
|
|
|
|||
|
|
def getsize(self, msg: str) -> Tuple[int, int]:
|
|||
|
|
"""
|
|||
|
|
获取文字在该图片 font_size 下所需要的空间
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
msg: 文本
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
Tuple[int, int]: 长宽
|
|||
|
|
"""
|
|||
|
|
return self.font.getsize(msg) # type: ignore
|
|||
|
|
|
|||
|
|
def __center_xy(
|
|||
|
|
self,
|
|||
|
|
pos: Tuple[int, int],
|
|||
|
|
width: int,
|
|||
|
|
height: int,
|
|||
|
|
center_type: CenterType | None,
|
|||
|
|
) -> Tuple[int, int]:
|
|||
|
|
"""
|
|||
|
|
根据居中类型定位xy
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
pos: 定位
|
|||
|
|
image: image
|
|||
|
|
center_type: 居中类型
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
Tuple[int, int]: 定位
|
|||
|
|
"""
|
|||
|
|
# _width, _height = pos
|
|||
|
|
if self.width and self.height:
|
|||
|
|
if center_type == "center":
|
|||
|
|
width = int((self.width - width) / 2)
|
|||
|
|
height = int((self.height - height) / 2)
|
|||
|
|
elif center_type == "width":
|
|||
|
|
width = int((self.width - width) / 2)
|
|||
|
|
height = pos[1]
|
|||
|
|
elif center_type == "height":
|
|||
|
|
width = pos[0]
|
|||
|
|
height = int((self.height - height) / 2)
|
|||
|
|
return width, height
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def paste(
|
|||
|
|
self,
|
|||
|
|
image: Self | tImage,
|
|||
|
|
pos: Tuple[int, int] = (0, 0),
|
|||
|
|
center_type: CenterType | None = None,
|
|||
|
|
) -> Self:
|
|||
|
|
"""贴图
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
image: BuildImage 或 Image
|
|||
|
|
pos: 定位.
|
|||
|
|
center_type: 居中.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
|
|||
|
|
异常:
|
|||
|
|
ValueError: 居中类型错误
|
|||
|
|
"""
|
|||
|
|
if center_type and center_type not in ["center", "height", "width"]:
|
|||
|
|
raise ValueError("center_type must be 'center', 'width' or 'height'")
|
|||
|
|
width, height = 0, 0
|
|||
|
|
_image = image
|
|||
|
|
if isinstance(image, BuildImage):
|
|||
|
|
_image = image.markImg
|
|||
|
|
if _image.width and _image.height and center_type:
|
|||
|
|
pos = self.__center_xy(pos, _image.width, _image.height, center_type)
|
|||
|
|
self.markImg.paste(_image, pos, _image) # type: ignore
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def point(
|
|||
|
|
self, pos: Tuple[int, int], fill: Tuple[int, int, int] | None = None
|
|||
|
|
) -> Self:
|
|||
|
|
"""
|
|||
|
|
绘制多个或单独的像素
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
pos: 坐标
|
|||
|
|
fill: 填充颜色.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.draw.point(pos, fill=fill)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def ellipse(
|
|||
|
|
self,
|
|||
|
|
pos: Tuple[int, int, int, int],
|
|||
|
|
fill: Tuple[int, int, int] | None = None,
|
|||
|
|
outline: Tuple[int, int, int] | None = None,
|
|||
|
|
width: int = 1,
|
|||
|
|
) -> Self:
|
|||
|
|
"""
|
|||
|
|
绘制圆
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
pos: 坐标范围
|
|||
|
|
fill: 填充颜色.
|
|||
|
|
outline: 描线颜色.
|
|||
|
|
width: 描线宽度.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.draw.ellipse(pos, fill, outline, width)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def text(
|
|||
|
|
self,
|
|||
|
|
pos: Tuple[int, int],
|
|||
|
|
text: str,
|
|||
|
|
fill: str | Tuple[int, int, int] = (0, 0, 0),
|
|||
|
|
center_type: CenterType | None = None,
|
|||
|
|
font: FreeTypeFont | str | Path | None = None,
|
|||
|
|
font_size: int = 10,
|
|||
|
|
) -> Self:
|
|||
|
|
"""
|
|||
|
|
在图片上添加文字
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
pos: 文字位置
|
|||
|
|
text: 文字内容
|
|||
|
|
fill: 文字颜色.
|
|||
|
|
center_type: 居中类型.
|
|||
|
|
font: 字体.
|
|||
|
|
font_size: 字体大小.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
|
|||
|
|
异常:
|
|||
|
|
ValueError: 居中类型错误
|
|||
|
|
"""
|
|||
|
|
if center_type and center_type not in ["center", "height", "width"]:
|
|||
|
|
raise ValueError("center_type must be 'center', 'width' or 'height'")
|
|||
|
|
width, height = 0, 0
|
|||
|
|
max_length_text = ""
|
|||
|
|
sentence = text.split("\n")
|
|||
|
|
for x in sentence:
|
|||
|
|
max_length_text = x if len(x) > len(max_length_text) else max_length_text
|
|||
|
|
if font:
|
|||
|
|
if not isinstance(font, FreeTypeFont):
|
|||
|
|
font = self.load_font(font, font_size)
|
|||
|
|
else:
|
|||
|
|
font = self.font
|
|||
|
|
if center_type:
|
|||
|
|
ttf_w, ttf_h = font.getsize(max_length_text) # type: ignore
|
|||
|
|
ttf_h = ttf_h * len(sentence)
|
|||
|
|
pos = self.__center_xy(pos, ttf_w, ttf_h, center_type)
|
|||
|
|
self.draw.text(pos, text, fill=fill, font=font)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def save(self, path: str | Path):
|
|||
|
|
"""
|
|||
|
|
保存图片
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
path: 图片路径
|
|||
|
|
"""
|
|||
|
|
self.markImg.save(path) # type: ignore
|
|||
|
|
|
|||
|
|
def show(self):
|
|||
|
|
"""
|
|||
|
|
说明:
|
|||
|
|
显示图片
|
|||
|
|
"""
|
|||
|
|
self.markImg.show()
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def resize(self, ratio: float = 0, width: int = 0, height: int = 0) -> Self:
|
|||
|
|
"""
|
|||
|
|
压缩图片
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
ratio: 压缩倍率.
|
|||
|
|
width: 压缩图片宽度至 width.
|
|||
|
|
height: 压缩图片高度至 height.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
|
|||
|
|
异常:
|
|||
|
|
ValueError: 缺少参数
|
|||
|
|
"""
|
|||
|
|
if not width and not height and not ratio:
|
|||
|
|
raise ValueError("缺少参数...")
|
|||
|
|
if self.width and self.height:
|
|||
|
|
if not width and not height and ratio:
|
|||
|
|
width = int(self.width * ratio)
|
|||
|
|
height = int(self.height * ratio)
|
|||
|
|
self.markImg = self.markImg.resize((width, height), Image.LANCZOS) # type: ignore
|
|||
|
|
self.width, self.height = self.markImg.size
|
|||
|
|
self.draw = ImageDraw.Draw(self.markImg)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def crop(self, box: Tuple[int, int, int, int]) -> Self:
|
|||
|
|
"""
|
|||
|
|
裁剪图片
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
box: 左上角坐标，右下角坐标 (left, upper, right, lower)
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.markImg = self.markImg.crop(box)
|
|||
|
|
self.width, self.height = self.markImg.size
|
|||
|
|
self.draw = ImageDraw.Draw(self.markImg)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def transparent(self, alpha_ratio: float = 1, n: int = 0) -> Self:
|
|||
|
|
"""
|
|||
|
|
图片透明化
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
alpha_ratio: 透明化程度.
|
|||
|
|
n: 透明化大小内边距.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.markImg = self.markImg.convert("RGBA")
|
|||
|
|
x, y = self.markImg.size
|
|||
|
|
for i in range(n, x - n):
|
|||
|
|
for k in range(n, y - n):
|
|||
|
|
color = self.markImg.getpixel((i, k))
|
|||
|
|
color = color[:-1] + (int(100 * alpha_ratio),)
|
|||
|
|
self.markImg.putpixel((i, k), color)
|
|||
|
|
self.draw = ImageDraw.Draw(self.markImg)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
def pic2bs4(self) -> str:
|
|||
|
|
"""
|
|||
|
|
BuildImage 转 base64
|
|||
|
|
"""
|
|||
|
|
buf = BytesIO()
|
|||
|
|
self.markImg.save(buf, format="PNG")
|
|||
|
|
base64_str = base64.b64encode(buf.getvalue()).decode()
|
|||
|
|
return "base64://" + base64_str
|
|||
|
|
|
|||
|
|
def convert(self, type_: ModeType) -> Self:
|
|||
|
|
"""
|
|||
|
|
修改图片类型
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
type_: ModeType
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.markImg = self.markImg.convert(type_)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def rectangle(
|
|||
|
|
self,
|
|||
|
|
xy: Tuple[int, int, int, int],
|
|||
|
|
fill: Tuple[int, int, int] | None = None,
|
|||
|
|
outline: str | None = None,
|
|||
|
|
width: int = 1,
|
|||
|
|
) -> Self:
|
|||
|
|
"""
|
|||
|
|
画框
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
xy: 坐标
|
|||
|
|
fill: 填充颜色.
|
|||
|
|
outline: 轮廓颜色.
|
|||
|
|
width: 线宽.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.draw.rectangle(xy, fill, outline, width)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def polygon(
|
|||
|
|
self,
|
|||
|
|
xy: List[Tuple[int, int]],
|
|||
|
|
fill: Tuple[int, int, int] = (0, 0, 0),
|
|||
|
|
outline: int = 1,
|
|||
|
|
) -> Self:
|
|||
|
|
"""
|
|||
|
|
画多边形
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
xy: 坐标
|
|||
|
|
fill: 颜色.
|
|||
|
|
outline: 线宽.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.draw.polygon(xy, fill, outline)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def line(
|
|||
|
|
self,
|
|||
|
|
xy: Tuple[int, int, int, int],
|
|||
|
|
fill: Tuple[int, int, int] | str = "#D8DEE4",
|
|||
|
|
width: int = 1,
|
|||
|
|
) -> Self:
|
|||
|
|
"""
|
|||
|
|
画线
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
xy: 坐标
|
|||
|
|
fill: 填充.
|
|||
|
|
width: 线宽.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.draw.line(xy, fill, width)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def circle(self) -> Self:
|
|||
|
|
"""
|
|||
|
|
图像变圆
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.markImg.convert("RGBA")
|
|||
|
|
size = self.markImg.size
|
|||
|
|
r2 = min(size[0], size[1])
|
|||
|
|
if size[0] != size[1]:
|
|||
|
|
self.markImg = self.markImg.resize((r2, r2), Image.LANCZOS) # type: ignore
|
|||
|
|
width = 1
|
|||
|
|
antialias = 4
|
|||
|
|
ellipse_box = [0, 0, r2 - 2, r2 - 2]
|
|||
|
|
mask = Image.new(
|
|||
|
|
size=[int(dim * antialias) for dim in self.markImg.size], # type: ignore
|
|||
|
|
mode="L",
|
|||
|
|
color="black",
|
|||
|
|
)
|
|||
|
|
draw = ImageDraw.Draw(mask)
|
|||
|
|
for offset, fill in (width / -2.0, "black"), (width / 2.0, "white"):
|
|||
|
|
left, top = [(value + offset) * antialias for value in ellipse_box[:2]]
|
|||
|
|
right, bottom = [(value - offset) * antialias for value in ellipse_box[2:]]
|
|||
|
|
draw.ellipse([left, top, right, bottom], fill=fill)
|
|||
|
|
mask = mask.resize(self.markImg.size, Image.LANCZOS)
|
|||
|
|
try:
|
|||
|
|
self.markImg.putalpha(mask)
|
|||
|
|
except ValueError:
|
|||
|
|
pass
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def circle_corner(
|
|||
|
|
self,
|
|||
|
|
radii: int = 30,
|
|||
|
|
point_list: List[Literal["lt", "rt", "lb", "rb"]] = ["lt", "rt", "lb", "rb"],
|
|||
|
|
) -> Self:
|
|||
|
|
"""
|
|||
|
|
矩形四角变圆
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
radii: 半径.
|
|||
|
|
point_list: 需要变化的角.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
# 画圆（用于分离4个角）
|
|||
|
|
img = self.markImg.convert("RGBA")
|
|||
|
|
alpha = img.split()[-1]
|
|||
|
|
circle = Image.new("L", (radii * 2, radii * 2), 0)
|
|||
|
|
draw = ImageDraw.Draw(circle)
|
|||
|
|
draw.ellipse((0, 0, radii * 2, radii * 2), fill=255) # 黑色方形内切白色圆形
|
|||
|
|
w, h = img.size
|
|||
|
|
if "lt" in point_list:
|
|||
|
|
alpha.paste(circle.crop((0, 0, radii, radii)), (0, 0))
|
|||
|
|
if "rt" in point_list:
|
|||
|
|
alpha.paste(circle.crop((radii, 0, radii * 2, radii)), (w - radii, 0))
|
|||
|
|
if "lb" in point_list:
|
|||
|
|
alpha.paste(circle.crop((0, radii, radii, radii * 2)), (0, h - radii))
|
|||
|
|
if "rb" in point_list:
|
|||
|
|
alpha.paste(
|
|||
|
|
circle.crop((radii, radii, radii * 2, radii * 2)),
|
|||
|
|
(w - radii, h - radii),
|
|||
|
|
)
|
|||
|
|
img.putalpha(alpha)
|
|||
|
|
self.markImg = img
|
|||
|
|
self.draw = ImageDraw.Draw(self.markImg)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def rotate(self, angle: int, expand: bool = False) -> Self:
|
|||
|
|
"""
|
|||
|
|
旋转图片
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
angle: 角度
|
|||
|
|
expand: 放大图片适应角度.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.markImg = self.markImg.rotate(angle, expand=expand)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def transpose(self, angle: Literal[0, 1, 2, 3, 4, 5, 6]) -> Self:
|
|||
|
|
"""
|
|||
|
|
旋转图片(包括边框)
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
angle: 角度
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
self.markImg.transpose(angle)
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
@run_sync
|
|||
|
|
def filter(self, filter_: str, aud: int | None = None) -> Self:
|
|||
|
|
"""
|
|||
|
|
图片变化
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
filter_: 变化效果
|
|||
|
|
aud: 利率.
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
BuildImage: Self
|
|||
|
|
"""
|
|||
|
|
_type = None
|
|||
|
|
if filter_ == "GaussianBlur": # 高斯模糊
|
|||
|
|
_type = ImageFilter.GaussianBlur
|
|||
|
|
elif filter_ == "EDGE_ENHANCE": # 锐化效果
|
|||
|
|
_type = ImageFilter.EDGE_ENHANCE
|
|||
|
|
elif filter_ == "BLUR": # 模糊效果
|
|||
|
|
_type = ImageFilter.BLUR
|
|||
|
|
elif filter_ == "CONTOUR": # 铅笔滤镜
|
|||
|
|
_type = ImageFilter.CONTOUR
|
|||
|
|
elif filter_ == "FIND_EDGES": # 边缘检测
|
|||
|
|
_type = ImageFilter.FIND_EDGES
|
|||
|
|
if _type:
|
|||
|
|
if aud:
|
|||
|
|
self.markImg = self.markImg.filter(_type(aud)) # type: ignore
|
|||
|
|
else:
|
|||
|
|
self.markImg = self.markImg.filter(_type)
|
|||
|
|
self.draw = ImageDraw.Draw(self.markImg)
|
|||
|
|
return self
|