publik-website/static/img/scripts/generate.py

48 lines
1.8 KiB
Python

#! /usr/bin/env python
import os
from optparse import OptionParser
import subprocess
import tempfile
import xml.etree.ElementTree as ET
import re
from PIL import Image
from PIL import PngImagePlugin
import sys
parser = OptionParser()
parser.add_option('-c', '--colour', dest='colour', metavar='COLOUR')
parser.add_option('-v', '--variant', dest='variant', metavar='VARIANT')
parser.add_option('-w', '--width', dest='width', metavar='WIDTH', default='42')
(options, args) = parser.parse_args()
for svg_path in args:
output_filename = svg_path.replace('.svg', '.png')
svg_path = os.path.join('sources', svg_path)
if options.variant:
output_filename = output_filename.replace('.png', '-%s.png' % options.variant)
author = None
if options.colour:
tree = ET.fromstring(file(svg_path).read().replace('#000000', '#%s' % options.colour))
for elem in tree.findall('*'):
if not elem.attrib.get('style'):
elem.attrib['style'] = 'fill:#%s' % options.colour
for elem in tree.getchildren():
if elem.tag == '{http://www.w3.org/2000/svg}text' and elem.text.startswith('Created by'):
author = elem.text[len('Created by')+1:]
tree.remove(elem)
for elem in tree.getchildren():
if elem.tag == '{http://www.w3.org/2000/svg}text' and 'Noun Project' in elem.text:
tree.remove(elem)
f = tempfile.NamedTemporaryFile(suffix='.svg', delete=False)
f.write(ET.tostring(tree))
f.close()
svg_path = f.name
# XXX: add author to png metadata
subprocess.call(['inkscape', '--without-gui',
'--file', svg_path,
'--export-area-drawing',
'--export-area-snap',
'--export-png', output_filename,
'--export-width', options.width])