|
@@ -1,82 +0,0 @@
|
|
|
-from sys import stdin
|
|
|
-from textwrap import fill, shorten
|
|
|
-
|
|
|
-from ujson import loads
|
|
|
-from PIL import Image, ImageDraw, ImageFont
|
|
|
-from emoji import demojize
|
|
|
-
|
|
|
-COLORS = [
|
|
|
- (0xee, 0x49, 0x28),
|
|
|
- (0x41, 0xa9, 0x03),
|
|
|
- (0xe0, 0x96, 0x02),
|
|
|
- (0x0f, 0x94, 0xed),
|
|
|
- (0x8f, 0x3b, 0xf7),
|
|
|
- (0xfc, 0x43, 0x80),
|
|
|
- (0x00, 0xa1, 0xc4),
|
|
|
- (0xeb, 0x70, 0x02)
|
|
|
-]
|
|
|
-
|
|
|
-data = loads(stdin.read())
|
|
|
-
|
|
|
-output_path = data['output_path']
|
|
|
-avatar_path = data['avatar_path']
|
|
|
-username = data['username']
|
|
|
-username_color = data['username_color']
|
|
|
-text = data['text']
|
|
|
-
|
|
|
-username = shorten(username, width=15)
|
|
|
-text = fill(text, width=30)
|
|
|
-
|
|
|
-username = demojize(username)
|
|
|
-text = demojize(text)
|
|
|
-
|
|
|
-margin = 24
|
|
|
-hpadding = 18
|
|
|
-vpadding = 14
|
|
|
-avatar_padding = 12
|
|
|
-avatar_dim = 64
|
|
|
-
|
|
|
-font = ImageFont.truetype('./resources/fonts/OpenSans-Regular.ttf', 24)
|
|
|
-username_font = ImageFont.truetype('./resources/fonts/OpenSans-Bold.ttf', 22)
|
|
|
-tail = Image.open('./resources/tail.png')
|
|
|
-mask = Image.new('L', (avatar_dim * 3, avatar_dim * 3), 0)
|
|
|
-draw = ImageDraw.Draw(mask)
|
|
|
-draw.ellipse((0, 0, avatar_dim * 3, avatar_dim * 3), fill=255)
|
|
|
-mask = mask.resize((avatar_dim, avatar_dim), Image.ANTIALIAS)
|
|
|
-tail_width, tail_height = tail.size
|
|
|
-
|
|
|
-avatar_width = avatar_dim + avatar_padding
|
|
|
-avatar_height = avatar_dim
|
|
|
-
|
|
|
-avatar = Image.open(avatar_path)
|
|
|
-avatar = avatar.resize((avatar_dim, avatar_dim))
|
|
|
-avatar.putalpha(mask)
|
|
|
-
|
|
|
-username_width, username_height = username_font.getsize(username)
|
|
|
-text_width, text_height = font.getsize_multiline(text)
|
|
|
-
|
|
|
-image_width = avatar_width + text_width + username_width + margin*2 + hpadding*2
|
|
|
-image_height = text_height + username_height + margin*2 + vpadding*2
|
|
|
-
|
|
|
-message = Image.new('RGBA', (image_width, image_height))
|
|
|
-message.paste(avatar, (margin, image_height - margin - avatar_height + 2), avatar)
|
|
|
-
|
|
|
-x1 = margin + avatar_dim + avatar_padding
|
|
|
-y1 = margin
|
|
|
-x2 = x1 + text_width + username_width + hpadding*2
|
|
|
-y2 = y1 + text_height + username_height + vpadding*2
|
|
|
-
|
|
|
-draw = ImageDraw.Draw(message)
|
|
|
-draw.rounded_rectangle((x1, y1, x2, y2), fill=(43, 43, 43), radius=7)
|
|
|
-draw.text((margin + avatar_width + hpadding, margin + vpadding/2), username, font=username_font, fill=COLORS[username_color])
|
|
|
-draw.multiline_text((margin + avatar_width + hpadding, margin + username_height + vpadding), text, font=font)
|
|
|
-
|
|
|
-message.paste(tail, (margin + avatar_width - tail_width + 8, image_height - margin - tail_height + 1), tail)
|
|
|
-
|
|
|
-ratio = image_width / image_height
|
|
|
-new_width = 512 if image_width >= image_height else min(int(512 / ratio), 512)
|
|
|
-new_height = 512 if image_height >= image_width else min(int(512 / ratio), 512)
|
|
|
-
|
|
|
-message = message.resize((new_width, new_height))
|
|
|
-
|
|
|
-message.save(output_path)
|