mirror of
https://github.com/Murasame-Dev/McStatus-Img.git
synced 2025-12-14 12:17:59 +00:00
123 lines
4.7 KiB
Python
123 lines
4.7 KiB
Python
from PIL import Image, ImageDraw, ImageFont, ImageFilter
|
|
from io import BytesIO
|
|
from PIL.ImageColor import getrgb
|
|
|
|
from .motd_formatter import foramt_motd
|
|
|
|
def load_font(font_url: str | None, font_size: int):
|
|
# 设置字体
|
|
if font_url == None:
|
|
font = ImageFont.load_default(font_size)
|
|
else:
|
|
font = ImageFont.truetype(font_url, font_size)
|
|
return font
|
|
|
|
def create_background(input: bytes, width: int, height: int):
|
|
background = Image.open(BytesIO(input))
|
|
w1, h1, w2, h2 = background.getbbox()
|
|
midw = (w1 + w2) // 2
|
|
midh = (h1 + h2) // 2
|
|
rheight = int(height / width * w2)
|
|
rwidth = w2
|
|
if rheight > h2:
|
|
rwidth = int(width / height * h2)
|
|
rheight = h2
|
|
|
|
background = background.crop((midw - int(rwidth / 2),
|
|
midh - int(rheight / 2),
|
|
midw + int(rwidth / 2),
|
|
midh + int(rheight / 2)))
|
|
background = background.resize((width, height),
|
|
Image.Resampling.LANCZOS)
|
|
|
|
blurred_background = background.filter(ImageFilter.GaussianBlur(radius=5))
|
|
|
|
image = blurred_background
|
|
return image
|
|
|
|
def draw_text_with_shadow(image: Image.Image,
|
|
text: str,
|
|
posx: int | float,
|
|
posy: int | float,
|
|
font_size: int,
|
|
font_url: str | None):
|
|
draw = ImageDraw.Draw(image)
|
|
font = load_font(font_url, font_size)
|
|
draw.text((posx + 2, posy + 2), text, font=font, fill='black')
|
|
draw.text((posx, posy), text, font=font, fill='white')
|
|
|
|
def draw_motd_text_with_shadow(image: Image.Image,
|
|
text: str,
|
|
posx: int | float,
|
|
posy: int | float,
|
|
font_size: int,
|
|
font_url: str | None):
|
|
draw = ImageDraw.Draw(image)
|
|
font = load_font(font_url, font_size)
|
|
motd_list = foramt_motd(text.strip(), draw, font)
|
|
for pos, color, text in motd_list:
|
|
draw.text((posx + pos + 1, posy + 1), text, font=font, fill='black')
|
|
draw.text((posx + pos, posy), text, font=font, fill=getrgb(color))
|
|
|
|
def create_image(background: bytes,
|
|
icon: str | None,
|
|
text_list: list[str],
|
|
motd_list: list[str],
|
|
font_url: str | None,
|
|
image_size: tuple[int, int] = (0, 0)):
|
|
# 图片尺寸
|
|
if type(image_size) != type(tuple()) or len(image_size) != 2:
|
|
assert ValueError("image_size is invalid")
|
|
|
|
if image_size[0] == 0 and image_size[1] == 0:
|
|
width = 1200
|
|
height = 400
|
|
else:
|
|
width = image_size[0]
|
|
height = image_size[1]
|
|
|
|
if (len(text_list) + len(motd_list)) * 20 + 20 > height:
|
|
height = (len(text_list) + len(motd_list)) * 20 + 20
|
|
font_size = height // 12
|
|
if font_size * 20 > width - width // 2.5:
|
|
font_size = (width - width // 2.5) // 20
|
|
|
|
try:
|
|
image = create_background(background, width, height)
|
|
except FileNotFoundError:
|
|
image = Image.new('RGB', (width, height), color='orange')
|
|
|
|
# 添加半透明蒙版层以增强文字可读性
|
|
overlay = Image.new('RGBA', (width, height), (0, 0, 0, 80)) # 半透明黑色蒙版
|
|
image.paste(overlay, (0, 0), overlay)
|
|
if width // 2 > height:
|
|
small_size = int(height * 0.8)
|
|
else:
|
|
small_size = width // 3
|
|
if icon == None:
|
|
small_image = Image.new('RGBA', (small_size, small_size), color='gray')
|
|
else:
|
|
small_image = Image.open(BytesIO(icon)).resize((small_size, small_size),
|
|
Image.Resampling.LANCZOS)
|
|
|
|
image.paste(small_image, (int(width / 3 - small_size), int(height / 2 - small_size / 2)))
|
|
|
|
text_list_size = len(text_list)
|
|
motd_list_size = len(motd_list)
|
|
start_posy = height / 2 - (text_list_size + motd_list_size) / 2 * font_size * 1.2
|
|
for i in range(motd_list_size):
|
|
draw_motd_text_with_shadow(image,
|
|
motd_list[i],
|
|
width / 2.5,
|
|
start_posy + font_size * 1.2 * i,
|
|
int(font_size * 0.8),
|
|
font_url)
|
|
for i in range(text_list_size):
|
|
draw_text_with_shadow(image,
|
|
text_list[i],
|
|
width / 2.5,
|
|
start_posy + font_size * 1.2 * (i + motd_list_size),
|
|
font_size,
|
|
font_url)
|
|
|
|
return image
|