| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182 |
- 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)
|
