pandora/scripts/poster.indiancinema.py
2018-06-19 08:33:26 +02:00

221 lines
6.8 KiB
Python
Executable file

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from __future__ import division
import os
from PIL import Image
from PIL import ImageDraw
import json
from optparse import OptionParser
import ox
from ox.image import getRGB, 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', ''))
director = ox.decode_html(u', '.join(data.get('director', [])))
year = str(data.get('year', ''))
frame = data.get('frame')
timeline = data.get('timeline')
poster_width = 704
poster_height = 1024
frame_height = 512
frame_ratio = poster_width / frame_height
timeline_height = 64
text_margin = 16
text_width = poster_width - 2 * text_margin
poster_image = Image.new('RGB', (poster_width, poster_height))
draw = ImageDraw.Draw(poster_image)
font_file = os.path.join(static_root, 'SourceSansProSemibold.ttf')
font_size = {'cinema': 117, 'director': 32, 'title': 48, 'year': 426}
font_lightness = {'cinema': 0.7, 'director': 0.8, 'title': 0.8, 'year': 0.6}
poster_lightness = {'image': 0.2, 'text': 0.4}
if year:
hue = int(year) % 100 / 100 * 360
saturation = 1
else:
hue = 0
saturation = 0
# background
draw.rectangle(
[(0, frame_height), (poster_width, poster_height - timeline_height)],
fill=getRGB((hue, saturation, poster_lightness['text']))
)
# year
if year:
drawText(
poster_image,
(-93, poster_height - timeline_height - font_size['year'] + 6),
year,
font_file,
font_size['year'],
getRGB((hue, saturation, font_lightness['year']))
)
# cinema
for y in [-1, 1]:
for x in [-1, 1]:
drawText(
poster_image,
(-10 + x, poster_height - timeline_height - font_size['cinema'] + 1 + y),
'Indiancine.ma',
font_file,
font_size['cinema'],
getRGB((hue, saturation, poster_lightness['text']))
)
drawText(
poster_image,
(-10, poster_height - timeline_height - font_size['cinema'] + 1),
'Indiancine.ma',
font_file,
font_size['cinema'],
getRGB((hue, saturation, font_lightness['cinema']))
)
# director and title
offset_top = frame_height + text_margin
if not director:
title_max_lines = 8
else:
title_max_lines = min(len(wrapText(
title,
text_width,
0,
font_file,
font_size['title']
)), 7)
director_max_lines = 11 - int((title_max_lines * 3 - 1) / 2)
# director
if director:
lines = wrapText(
director,
text_width,
director_max_lines,
font_file,
font_size['director']
)
for line in lines:
for y in [-1, 1]:
for x in [-1, 1]:
drawText(
poster_image,
(text_margin + x, offset_top + y),
line,
font_file,
font_size['director'],
getRGB((hue, saturation, poster_lightness['text']))
)
size = drawText(
poster_image,
(text_margin, offset_top),
line,
font_file,
font_size['director'],
getRGB((hue, saturation, font_lightness['director']))
)
offset_top += font_size['director'] + 2
offset_top += size[1] - font_size['director'] - 2
# title
lines = wrapText(
title,
text_width,
title_max_lines,
font_file,
font_size['title']
)
for line in lines:
for y in [-1, 1]:
for x in [-1, 1]:
drawText(
poster_image,
(text_margin + x, offset_top + y),
line,
font_file,
font_size['title'],
getRGB((hue, saturation, poster_lightness['text']))
)
drawText(
poster_image,
(text_margin, offset_top),
line,
font_file,
font_size['title'],
getRGB((hue, saturation, font_lightness['title']))
)
offset_top += font_size['title'] + 3
# frame
if frame:
frame_image = Image.open(frame)
frame_image_ratio = frame_image.size[0] / frame_image.size[1]
if frame_ratio < frame_image_ratio:
frame_image = frame_image.resize(
(int(frame_height * frame_image_ratio), frame_height),
Image.ANTIALIAS
)
left = int((frame_image.size[0] - poster_width) / 2)
frame_image = frame_image.crop(
(left, 0, left + poster_width, frame_height)
)
else:
frame_image = frame_image.resize(
(poster_width, int(poster_width / frame_image_ratio)),
Image.ANTIALIAS
)
top = int((frame_image.size[1] - frame_height) / 2)
frame_image = frame_image.crop(
(0, top, poster_width, top + frame_height)
)
poster_image.paste(frame_image, (0, 0))
else:
draw.rectangle(
[(0, 0), (poster_width, frame_height)],
fill=getRGB((hue, saturation, poster_lightness['image']))
)
# timeline
if timeline:
timeline_image = Image.open(timeline)
timeline_image = timeline_image.resize(
(poster_width, timeline_height),
Image.ANTIALIAS
)
poster_image.paste(timeline_image, (0, poster_height - timeline_height))
else:
draw.rectangle(
[(0, poster_height - timeline_height), (poster_width, poster_height)],
fill=getRGB((hue, saturation, poster_lightness['image']))
)
poster_image.save(poster, quality=100)
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()