mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
97 lines
2.6 KiB
Python
97 lines
2.6 KiB
Python
from PIL import Image, ImageDraw, ImageFont
|
|
from pathlib import Path
|
|
|
|
res_path = Path(__file__).parent.parent / 'res'
|
|
font = ImageFont.truetype(str(res_path / 'wqy-microhei.ttc'), 20)
|
|
|
|
width = 500
|
|
|
|
color = [
|
|
{'front': 'black', 'back': 'white'},
|
|
{'front': 'white', 'back': 'ForestGreen'},
|
|
{'front': 'white', 'back': 'DarkOrange'},
|
|
{'front': 'white', 'back': 'BlueViolet'},
|
|
]
|
|
|
|
|
|
def create_image(item_number):
|
|
height = item_number * 30
|
|
im = Image.new('RGBA', (width, height), (255, 255, 255, 0))
|
|
return im
|
|
|
|
|
|
def draw_rec(im, color, x, y, w, h, r):
|
|
draw = ImageDraw.Draw(im)
|
|
draw.rectangle((x+r, y, x+w-r, y+h), fill=color)
|
|
draw.rectangle((x, y+r, x+w, y+h-r), fill=color)
|
|
r = r * 2
|
|
draw.ellipse((x, y, x+r, y+r), fill=color)
|
|
draw.ellipse((x+w-r, y, x+w, y+r), fill=color)
|
|
draw.ellipse((x, y+h-r, x+r, y+h), fill=color)
|
|
draw.ellipse((x+w-r, y+h-r, x+w, y+h), fill=color)
|
|
|
|
|
|
def draw_text(im, x, y, w, h, text, align, color):
|
|
draw = ImageDraw.Draw(im)
|
|
tw, th = draw.textsize(text, font=font)
|
|
y = y + (h - th) / 2
|
|
if align == 0: # 居中
|
|
x = x + (w - tw) / 2
|
|
elif align == 1: # 左对齐
|
|
x = x + 5
|
|
elif align == 2: # 右对齐
|
|
x = x + w - tw - 5
|
|
draw.text((x, y), text, fill=color, font=font)
|
|
|
|
|
|
def draw_item(im, n, t, text, days, forever):
|
|
if t >= len(color):
|
|
t = 1
|
|
x = 0
|
|
y = n * 30
|
|
height = 28
|
|
|
|
draw_rec(im, color[t]['back'], x, y, width, height, 6)
|
|
|
|
im1 = Image.new('RGBA', (width - 120, 28), (255, 255, 255, 0))
|
|
draw_text(im1, 0, 0, width, height, text, 1, color[t]['front'])
|
|
_, _, _, a = im1.split()
|
|
im.paste(im1, (x, y), mask=a)
|
|
|
|
if days > 0:
|
|
if forever:
|
|
text1 = '永久开放'
|
|
else:
|
|
text1 = f'{days}天后结束'
|
|
elif days < 0:
|
|
text1 = f'{-days}天后开始'
|
|
else:
|
|
text1 = '即将结束'
|
|
draw_text(im, x, y, width, height, text1, 2, color[t]['front'])
|
|
|
|
|
|
def draw_title(im, n, left=None, middle=None, right=None):
|
|
x = 0
|
|
y = n * 30
|
|
height = 28
|
|
|
|
draw_rec(im, color[0]['back'], x, y, width, height, 6)
|
|
if middle:
|
|
draw_text(im, x, y, width, height, middle, 0, color[0]['front'])
|
|
if left:
|
|
draw_text(im, x, y, width, height, left, 1, color[0]['front'])
|
|
if right:
|
|
draw_text(im, x, y, width, height, right, 2, color[0]['front'])
|
|
|
|
|
|
def draw_title1(im, n, day_list):
|
|
x = 0
|
|
y = n * 30
|
|
height = 28
|
|
color = 'black'
|
|
|
|
n = len(day_list)
|
|
for i in range(n):
|
|
x = width / n * i
|
|
draw_text(im, x, y, width, height, day_list[i], 1, color)
|