pandora/scripts/poster.padma.py
2023-08-24 23:39:54 +02:00

81 lines
2.8 KiB
Python
Executable file

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
from PIL import Image
from PIL import ImageDraw
import json
from optparse import OptionParser
import ox
from ox.image import drawText, wrapText
import sys
root_dir = os.path.normpath(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
static_root = os.path.join(os.path.dirname(__file__), 'data')
def render_poster(data, poster):
title = ox.decode_html(data.get('title', ''))
id = data['id']
frame = data.get('frame')
timeline = data.get('timeline')
poster_width = 640
poster_height = 1024
poster_ratio = poster_width / poster_height
poster_color = (255, 255, 0)
poster_image = Image.new('RGB', (poster_width, poster_height))
font_file = os.path.join(static_root, 'DejaVuSansCondensedBold.ttf')
font_size = 48
# timeline
timeline_height = 64
timeline_lines = 16
if timeline:
timeline_image = Image.open(timeline)
timeline_image = timeline_image.resize((10240, timeline_height), Image.LANCZOS)
for i in range(timeline_lines):
line_image = timeline_image.crop((i * poster_width, 0, (i + 1) * poster_width, 64))
poster_image.paste(line_image, (0, i * timeline_height))
# id
text = 'Pad.ma/' + id
text_image = Image.new('RGB', (1, 1))
text_size = drawText(text_image, (0, 0), text, font_file, font_size, poster_color)
text_width = poster_width
text_height = timeline_height
text_left = int((poster_width - text_width) / 2)
text_top = 14 * timeline_height
for y in range(text_top, text_top + text_height):
for x in range(text_left, text_left + text_width):
if y < text_top + 4 or y >= text_top + text_height - 4:
poster_image.putpixel((x, y), poster_color)
else:
pixel = list(poster_image.getpixel((x, y)))
for c in range(3):
pixel[c] = int((pixel[c] + poster_color[c]) / 4)
poster_image.putpixel((x, y), tuple(pixel))
drawText(poster_image, ((poster_width - text_size[0]) / 2, text_top + (text_height - text_size[1]) / 2), text, font_file, font_size, poster_color)
poster_image.save(poster)
def main():
parser = OptionParser()
parser.add_option('-d', '--data', dest='data', help='json file with metadata', default=None)
parser.add_option('-p', '--poster', dest='poster', help='Poster (image file to be written)')
(options, args) = parser.parse_args()
if None in (options.data, options.poster):
parser.print_help()
sys.exit()
if options.data == '-':
data = json.load(sys.stdin)
else:
with open(options.data) as f:
data = json.load(f)
render_poster(data, options.poster)
if __name__ == "__main__":
main()