1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
#!/usr/bin/env python2.4
import errno
import os
from StringIO import StringIO
import sys
from docutils.core import publish_file
from docutils.parsers import rst
from elementtree.ElementTree import XML
from elementtree import HTMLTreeBuilder
import kid
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)
args = sys.argv[1:]
assert len(args) > 0
if len(args) > 1:
rest_file = safe_open(args[1], 'r')
else:
rest_file = sys.stdin
if len(args) > 2:
out_file = safe_open(args[2], 'w')
else:
out_file = sys.stdout
out_file.write(kidified_rest(rest_file, args[0]))
|