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
|
from __future__ import division
import sys
from os.path import basename
from docutils import nodes
from docutils import statemachine
try:
from sphinx.util.compat import Directive
except ImportError:
from docutils.parsers.rst import Directive
try:
from StringIO import StringIO
except ImportError:
from io import StringIO
class ExecDirective(Directive):
"""Execute the specified python code and insert the output into the document"""
has_content = True
def run(self):
old_stdout, sys.stdout = sys.stdout, StringIO()
tab_width = self.options.get('tab-width', self.state.document.settings.tab_width)
source = self.state_machine.input_lines.source(self.lineno - self.state_machine.input_offset - 1)
try:
exec('\n'.join(self.content), globals())
text = sys.stdout.getvalue()
lines = statemachine.string2lines(text, tab_width, convert_whitespace = True)
self.state_machine.insert_input(lines, source)
return []
except Exception:
return [nodes.error(None, nodes.paragraph(text = "Unable to execute python code at %s:%d:" % (basename(source), self.lineno)), nodes.paragraph(text = str(sys.exc_info()[1])))]
finally:
sys.stdout = old_stdout
def setup(app):
app.add_directive('exec', ExecDirective)
|
