bak.ma config

scripts/poster.py

@@ -0,0 +1,86 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+# vi:si:et:sw=4:sts=4:ts=4
+from __future__ import division
+import os
+
+root_dir = os.path.normpath(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+
+# using virtualenv's activate_this.py to reorder sys.path
+activate_this = os.path.join(root_dir, 'bin', 'activate_this.py')
+execfile(activate_this, dict(__file__=activate_this))
+
+import Image
+import ImageDraw
+import json
+from optparse import OptionParser
+import ox
+from ox.image import drawText, wrapText
+import sys
+
+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.ANTIALIAS)
+        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 = 'Bak.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()
+