~ubuntu-branches/ubuntu/lucid/bzr/lucid-proposed

#!/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)