62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
from docutils import writers
|
|
from docutils import nodes
|
|
|
|
class LitreTranslator(nodes.GenericNodeVisitor):
|
|
|
|
def __init__(self, document, config):
|
|
nodes.GenericNodeVisitor.__init__(self,document)
|
|
self._config = config
|
|
|
|
def default_visit(self, node):
|
|
pass
|
|
# print '**visiting:', repr(node)
|
|
|
|
def default_departure(self, node):
|
|
pass
|
|
# print '**departing:', repr(node)
|
|
|
|
def visit_raw(self, node):
|
|
if node.has_key('format'):
|
|
key = node['format'].lower()
|
|
if key == 'litre':
|
|
# This is probably very evil ;-)
|
|
#if node.has_key('source'):
|
|
# node.file = node.attributes['source']
|
|
|
|
self._handle_code(node, node.astext())
|
|
|
|
raise nodes.SkipNode
|
|
|
|
def visit_comment(self, node):
|
|
code = node.astext()
|
|
if code[0] == '@':
|
|
self._handle_code(node, code[1:].strip())
|
|
|
|
def _handle_code(self, node, code):
|
|
start_line = node.line or 0
|
|
start_line -= code.count('\n') + 2 # docutils bug workaround?
|
|
try:
|
|
self._execute(compile( start_line*'\n' + code, str(node.source), 'exec'))
|
|
except:
|
|
print '\n------- begin offending Python source -------'
|
|
print code
|
|
print '------- end offending Python source -------'
|
|
raise
|
|
|
|
def _execute(self, code):
|
|
"""Override this to set up local variable context for code before
|
|
invoking it
|
|
"""
|
|
eval(code)
|
|
|
|
class Writer(writers.Writer):
|
|
translator = LitreTranslator
|
|
_config = None
|
|
|
|
def translate(self):
|
|
visitor = self.translator(self.document, self._config)
|
|
self.document.walkabout(visitor)
|
|
self.output = visitor.astext()
|
|
|
|
|