✨ 新增utils.image方法

gsuid_core/utils/image/convert.py

@@ -0,0 +1,109 @@
+from io import BytesIO
+from pathlib import Path
+from base64 import b64encode
+from typing import Union, overload
+
+import aiofiles
+from PIL import Image, ImageFont
+
+
+@overload
+async def convert_img(img: Image.Image, is_base64: bool = False) -> bytes:
+    ...
+
+
+@overload
+async def convert_img(img: Image.Image, is_base64: bool = True) -> str:
+    ...
+
+
+@overload
+async def convert_img(img: bytes, is_base64: bool = False) -> str:
+    ...
+
+
+@overload
+async def convert_img(img: Path, is_base64: bool = False) -> str:
+    ...
+
+
+async def convert_img(
+    img: Union[Image.Image, str, Path, bytes], is_base64: bool = False
+):
+    """
+    :说明:
+      将PIL.Image对象转换为bytes或者base64格式。
+    :参数:
+      * img (Image): 图片。
+      * is_base64 (bool): 是否转换为base64格式, 不填默认转为bytes。
+    :返回:
+      * res: bytes对象或base64编码图片。
+    """
+    if isinstance(img, Image.Image):
+        img = img.convert('RGB')
+        result_buffer = BytesIO()
+        img.save(result_buffer, format='PNG', quality=80, subsampling=0)
+        res = result_buffer.getvalue()
+        if is_base64:
+            res = 'base64://' + b64encode(res).decode()
+        return res
+    elif isinstance(img, bytes):
+        pass
+    else:
+        async with aiofiles.open(img, 'rb') as fp:
+            img = await fp.read()
+    return f'base64://{b64encode(img).decode()}'
+
+
+async def str_lenth(r: str, size: int, limit: int = 540) -> str:
+    result = ''
+    temp = 0
+    for i in r:
+        if i == '\n':
+            temp = 0
+            result += i
+            continue
+
+        if temp >= limit:
+            result += '\n' + i
+            temp = 0
+        else:
+            result += i
+
+        if i.isdigit():
+            temp += round(size / 10 * 6)
+        elif i == '/':
+            temp += round(size / 10 * 2.2)
+        elif i == '.':
+            temp += round(size / 10 * 3)
+        elif i == '%':
+            temp += round(size / 10 * 9.4)
+        else:
+            temp += size
+    return result
+
+
+def get_str_size(
+    r: str, font: ImageFont.FreeTypeFont, limit: int = 540
+) -> str:
+    result = ''
+    line = ''
+    for i in r:
+        if i == '\n':
+            result += f'{line}\n'
+            line = ''
+            continue
+
+        line += i
+        size, _ = font.getsize(line)
+        if size >= limit:
+            result += f'{line}\n'
+            line = ''
+    else:
+        result += line
+    return result
+
+
+def get_height(content: str, size: int) -> int:
+    line_count = content.count('\n')
+    return (line_count + 1) * size

gsuid_core/utils/image/image_tools.py

@@ -0,0 +1,312 @@
+import math
+import random
+from io import BytesIO
+from pathlib import Path
+from typing import Tuple, Union, Optional
+
+import httpx
+from httpx import get
+from PIL import Image, ImageDraw, ImageFont
+
+TEXT_PATH = Path(__file__).parent / 'texture2d'
+
+
+async def get_pic(url, size: Optional[Tuple[int, int]] = None) -> Image.Image:
+    """
+    从网络获取图片, 格式化为RGBA格式的指定尺寸
+    """
+    async with httpx.AsyncClient(timeout=None) as client:
+        resp = await client.get(url=url)
+        if resp.status_code != 200:
+            if size is None:
+                size = (960, 600)
+            return Image.new('RGBA', size)
+        pic = Image.open(BytesIO(resp.read()))
+        pic = pic.convert("RGBA")
+        if size is not None:
+            pic = pic.resize(size, Image.LANCZOS)
+        return pic
+
+
+def draw_text_by_line(
+    img: Image.Image,
+    pos: Tuple[int, int],
+    text: str,
+    font: ImageFont.FreeTypeFont,
+    fill: Union[Tuple[int, int, int, int], str],
+    max_length: float,
+    center=False,
+    line_space: Optional[float] = None,
+):
+    """
+    在图片上写长段文字, 自动换行
+    max_length单行最大长度, 单位像素
+    line_space  行间距, 单位像素, 默认是字体高度的0.3倍
+    """
+    x, y = pos
+    _, h = font.getsize('X')
+    if line_space is None:
+        y_add = math.ceil(1.3 * h)
+    else:
+        y_add = math.ceil(h + line_space)
+    draw = ImageDraw.Draw(img)
+    row = ""  # 存储本行文字
+    length = 0  # 记录本行长度
+    for character in text:
+        w, h = font.getsize(character)  # 获取当前字符的宽度
+        if length + w * 2 <= max_length:
+            row += character
+            length += w
+        else:
+            row += character
+            if center:
+                font_size = font.getsize(row)
+                x = math.ceil((img.size[0] - font_size[0]) / 2)
+            draw.text((x, y), row, font=font, fill=fill)
+            row = ""
+            length = 0
+            y += y_add
+    if row != "":
+        if center:
+            font_size = font.getsize(row)
+            x = math.ceil((img.size[0] - font_size[0]) / 2)
+        draw.text((x, y), row, font=font, fill=fill)
+
+
+def easy_paste(
+    im: Image.Image, im_paste: Image.Image, pos=(0, 0), direction="lt"
+):
+    """
+    inplace method
+    快速粘贴, 自动获取被粘贴图像的坐标。
+    pos应当是粘贴点坐标，direction指定粘贴点方位，例如lt为左上
+    """
+    x, y = pos
+    size_x, size_y = im_paste.size
+    if "d" in direction:
+        y = y - size_y
+    if "r" in direction:
+        x = x - size_x
+    if "c" in direction:
+        x = x - int(0.5 * size_x)
+        y = y - int(0.5 * size_y)
+    im.paste(im_paste, (x, y, x + size_x, y + size_y), im_paste)
+
+
+def easy_alpha_composite(
+    im: Image.Image, im_paste: Image.Image, pos=(0, 0), direction="lt"
+) -> Image.Image:
+    '''
+    透明图像快速粘贴
+    '''
+    base = Image.new("RGBA", im.size)
+    easy_paste(base, im_paste, pos, direction)
+    base = Image.alpha_composite(im, base)
+    return base
+
+
+async def get_qq_avatar(
+    qid: Optional[Union[int, str]] = None, avatar_url: Optional[str] = None
+) -> Image.Image:
+    if qid:
+        avatar_url = f'http://q1.qlogo.cn/g?b=qq&nk={qid}&s=640'
+    elif avatar_url is None:
+        avatar_url = 'https://q1.qlogo.cn/g?b=qq&nk=3399214199&s=640'
+    char_pic = Image.open(BytesIO(get(avatar_url).content)).convert('RGBA')
+    return char_pic
+
+
+async def draw_pic_with_ring(
+    pic: Image.Image,
+    size: int,
+    bg_color: Optional[Tuple[int, int, int]] = None,
+):
+    '''
+    :说明:
+      绘制一张带白色圆环的1:1比例图片。
+
+    :参数:
+      * pic: `Image.Image`: 要修改的图片。
+      * size: `int`: 最后传出图片的大小(1:1)。
+      * bg_color: `Optional[Tuple[int, int, int]]`: 是否指定圆环内背景颜色。
+
+    :返回:
+      * img: `Image.Image`: 图片对象
+    '''
+    ring_pic = Image.open(TEXT_PATH / 'ring.png')
+    mask_pic = Image.open(TEXT_PATH / 'mask.png')
+    img = Image.new('RGBA', (size, size))
+    mask = mask_pic.resize((size, size))
+    ring = ring_pic.resize((size, size))
+    resize_pic = crop_center_img(pic, size, size)
+    if bg_color:
+        img_color = Image.new('RGBA', (size, size), bg_color)
+        img_color.paste(resize_pic, (0, 0), resize_pic)
+        img.paste(img_color, (0, 0), mask)
+    else:
+        img.paste(resize_pic, (0, 0), mask)
+    img.paste(ring, (0, 0), ring)
+    return img
+
+
+def crop_center_img(
+    img: Image.Image, based_w: int, based_h: int
+) -> Image.Image:
+    # 确定图片的长宽
+    based_scale = '%.3f' % (based_w / based_h)
+    w, h = img.size
+    scale_f = '%.3f' % (w / h)
+    new_w = math.ceil(based_h * float(scale_f))
+    new_h = math.ceil(based_w / float(scale_f))
+    if scale_f > based_scale:
+        resize_img = img.resize((new_w, based_h), Image.ANTIALIAS)
+        x1 = int(new_w / 2 - based_w / 2)
+        y1 = 0
+        x2 = int(new_w / 2 + based_w / 2)
+        y2 = based_h
+    else:
+        resize_img = img.resize((based_w, new_h), Image.ANTIALIAS)
+        x1 = 0
+        y1 = int(new_h / 2 - based_h / 2)
+        x2 = based_w
+        y2 = int(new_h / 2 + based_h / 2)
+    crop_img = resize_img.crop((x1, y1, x2, y2))
+    return crop_img
+
+
+class CustomizeImage:
+    def __init__(self, bg_path: Path) -> None:
+        self.bg_path = bg_path
+
+    def get_image(
+        self, image: Union[str, Image.Image], based_w: int, based_h: int
+    ) -> Image.Image:
+        # 获取背景图片
+        if isinstance(image, Image.Image):
+            edit_bg = image
+        elif image:
+            edit_bg = Image.open(BytesIO(get(image).content)).convert('RGBA')
+        else:
+            path = random.choice(list(self.bg_path.iterdir()))
+            edit_bg = Image.open(path).convert('RGBA')
+
+        # 确定图片的长宽
+        bg_img = crop_center_img(edit_bg, based_w, based_h)
+        return bg_img
+
+    @staticmethod
+    def get_dominant_color(pil_img: Image.Image) -> Tuple[int, int, int]:
+        img = pil_img.copy()
+        img = img.convert("RGBA")
+        img = img.resize((1, 1), resample=0)
+        dominant_color = img.getpixel((0, 0))
+        return dominant_color
+
+    @staticmethod
+    def get_bg_color(
+        edit_bg: Image.Image, is_light: Optional[bool] = False
+    ) -> Tuple[int, int, int]:
+        # 获取背景主色
+        color = 8
+        q = edit_bg.quantize(colors=color, method=2)
+        bg_color = (0, 0, 0)
+        if is_light:
+            based_light = 195
+        else:
+            based_light = 120
+        temp = 9999
+        for i in range(color):
+            bg = tuple(
+                q.getpalette()[  # type:ignore
+                    i * 3 : (i * 3) + 3  # noqa:E203
+                ]
+            )
+            light_value = bg[0] * 0.3 + bg[1] * 0.6 + bg[2] * 0.1
+            if abs(light_value - based_light) < temp:  # noqa:E203
+                bg_color = bg
+                temp = abs(light_value - based_light)
+        return bg_color
+
+    @staticmethod
+    def get_text_color(bg_color: Tuple[int, int, int]) -> Tuple[int, int, int]:
+        # 通过背景主色（bg_color）确定文字主色
+        r = 125
+        if max(*bg_color) > 255 - r:
+            r *= -1
+        text_color = (
+            math.floor(bg_color[0] + r if bg_color[0] + r <= 255 else 255),
+            math.floor(bg_color[1] + r if bg_color[1] + r <= 255 else 255),
+            math.floor(bg_color[2] + r if bg_color[2] + r <= 255 else 255),
+        )
+        return text_color
+
+    @staticmethod
+    def get_char_color(bg_color: Tuple[int, int, int]) -> Tuple[int, int, int]:
+        r = 140
+        if max(*bg_color) > 255 - r:
+            r *= -1
+        char_color = (
+            math.floor(bg_color[0] + 5 if bg_color[0] + r <= 255 else 255),
+            math.floor(bg_color[1] + 5 if bg_color[1] + r <= 255 else 255),
+            math.floor(bg_color[2] + 5 if bg_color[2] + r <= 255 else 255),
+        )
+        return char_color
+
+    @staticmethod
+    def get_char_high_color(
+        bg_color: Tuple[int, int, int]
+    ) -> Tuple[int, int, int]:
+        r = 140
+        d = 20
+        if max(*bg_color) > 255 - r:
+            r *= -1
+        char_color = (
+            math.floor(bg_color[0] + d if bg_color[0] + r <= 255 else 255),
+            math.floor(bg_color[1] + d if bg_color[1] + r <= 255 else 255),
+            math.floor(bg_color[2] + d if bg_color[2] + r <= 255 else 255),
+        )
+        return char_color
+
+    @staticmethod
+    def get_bg_detail_color(
+        bg_color: Tuple[int, int, int]
+    ) -> Tuple[int, int, int]:
+        r = 140
+        if max(*bg_color) > 255 - r:
+            r *= -1
+        bg_detail_color = (
+            math.floor(bg_color[0] - 20 if bg_color[0] + r <= 255 else 255),
+            math.floor(bg_color[1] - 20 if bg_color[1] + r <= 255 else 255),
+            math.floor(bg_color[2] - 20 if bg_color[2] + r <= 255 else 255),
+        )
+        return bg_detail_color
+
+    @staticmethod
+    def get_highlight_color(
+        color: Tuple[int, int, int]
+    ) -> Tuple[int, int, int]:
+        red_color = color[0]
+        green_color = color[1]
+        blue_color = color[2]
+
+        highlight_color = {
+            'red': red_color - 127 if red_color > 127 else 127,
+            'green': green_color - 127 if green_color > 127 else 127,
+            'blue': blue_color - 127 if blue_color > 127 else 127,
+        }
+
+        max_color = max(highlight_color.values())
+
+        name = ''
+        for _highlight_color in highlight_color:
+            if highlight_color[_highlight_color] == max_color:
+                name = str(_highlight_color)
+
+        if name == 'red':
+            return red_color, highlight_color['green'], highlight_color['blue']
+        elif name == 'green':
+            return highlight_color['red'], green_color, highlight_color['blue']
+        elif name == 'blue':
+            return highlight_color['red'], highlight_color['green'], blue_color
+        else:
+            return 0, 0, 0  # Error