HeXi/hexi/plugins/nonebot_plugin_helldivers_tools/utils/image_builder.py

import os
import random
from io import BytesIO
from datetime import datetime
from PIL import Image, ImageOps, ImageDraw, ImageFont

basic_path = os.path.dirname(os.path.dirname(__file__))
save_path = os.path.join(basic_path, "temp")
img_path = os.path.join(basic_path, "res", "img")
data_path = os.path.join(basic_path, "data")


def create_image(selected_equipment):
    new_img = Image.new('RGBA', (800, 500), (0, 0, 0, 1000))
    logo_path = img_path + "/super earth.png"
    logo = Image.open(logo_path)
    new_img = image_paste(logo, new_img, (682, 20))
    draw = ImageDraw.Draw(new_img)
    ch_text_font = ImageFont.truetype(data_path + '/font/msyh.ttc', 36)
    pos_horizon = 20

    for equipment in selected_equipment:
        name = equipment['name']
        alias = random.choice(equipment['alias']) if equipment['alias'] else None

        path = basic_path + "/" + equipment['path'].replace("\\", "/")
        icon = Image.open(path)
        icon = icon.resize((100, 100))
        new_img = image_paste(icon, new_img, (20, pos_horizon))
        draw.text((140, pos_horizon + 5), '战备名称：', fill='white', font=ch_text_font)
        draw.text((140, pos_horizon + 50), f'{name} - ({alias})', fill='white', font=ch_text_font)
        pos_horizon += 120

    b_io = BytesIO()
    new_img = ImageOps.expand(new_img, border=10, fill="white")
    new_img.save(b_io, format="PNG")
    return b_io


def image_paste(paste_image, under_image, pos):
    if paste_image.mode == 'RGBA':
        under_image.paste(paste_image, pos, mask=paste_image.split()[3])
    else:
        under_image.paste(paste_image, pos)
    return under_image