#!/usr/bin/env python
import errno
import os
from StringIO import StringIO
import sys
try:
from docutils.core import publish_file
from docutils.parsers import rst
except ImportError:
print "Missing dependency. Please install docutils."
sys.exit(1)
try:
from elementtree.ElementTree import XML
from elementtree import HTMLTreeBuilder
except ImportError:
print "Missing dependency. Please install ElementTree."
sys.exit(1)
try:
import kid
except ImportError:
print "Missing dependency. Please install Kid."
sys.exit(1)
def kidified_rest(rest_file, template_name):
xhtml_file = StringIO()
# prevent docutils from autoclosing the StringIO
xhtml_file.close = lambda: None
xhtml = publish_file(rest_file, writer_name='html', destination=xhtml_file,
settings_overrides={"doctitle_xform": 0}
)
xhtml_file.seek(0)
xml = HTMLTreeBuilder.parse(xhtml_file)
head = xml.find('head')
body = xml.find('body')
assert head is not None
assert body is not None
template=kid.Template(file=template_name,
head=head, body=body)
return (template.serialize(output="html"))
def safe_open(filename, mode):
try:
return open(filename, mode + 'b')
except IOError, e:
if e.errno != errno.ENOENT:
raise
sys.stderr.write('file not found: %s\n' % sys.argv[2])
sys.exit(3)
def main(template, source=None, target=None):
if source is not None:
rest_file = safe_open(source, 'r')
else:
rest_file = sys.stdin
if target is not None:
out_file = safe_open(target, 'w')
else:
out_file = sys.stdout
out_file.write(kidified_rest(rest_file, template))
assert len(sys.argv) > 1
# Strip options so only the arguments are passed
args = [x for x in sys.argv[1:] if not x.startswith('-')]
main(*args)
|