mirror of
https://github.com/triqs/dft_tools
synced 2024-12-23 12:55:17 +01:00
b5a85e3034
Co-authored-by: Dylan Simon <dylan@dylex.net> Co-authored-by: Alexander Hampel <ahampel@flatironinstitute.org>
94 lines
2.6 KiB
Python
94 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
sphinxcontirb.autorun
|
|
~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Run the code and insert stdout after the code block.
|
|
"""
|
|
import os
|
|
from subprocess import PIPE, Popen
|
|
|
|
from docutils import nodes
|
|
from docutils.parsers.rst import Directive, directives
|
|
from sphinx.errors import SphinxError
|
|
|
|
from sphinx_autorun import version
|
|
|
|
__version__ = version.version
|
|
|
|
|
|
class RunBlockError(SphinxError):
|
|
category = 'runblock error'
|
|
|
|
|
|
class AutoRun(object):
|
|
here = os.path.abspath(__file__)
|
|
pycon = os.path.join(os.path.dirname(here), 'pycon.py')
|
|
config = {
|
|
'pycon': 'python ' + pycon,
|
|
'pycon_prefix_chars': 4,
|
|
'pycon_show_source': False,
|
|
'console': 'bash',
|
|
'console_prefix_chars': 1,
|
|
}
|
|
|
|
@classmethod
|
|
def builder_init(cls, app):
|
|
cls.config.update(app.builder.config.autorun_languages)
|
|
|
|
|
|
class RunBlock(Directive):
|
|
has_content = True
|
|
required_arguments = 1
|
|
optional_arguments = 0
|
|
final_argument_whitespace = False
|
|
option_spec = {
|
|
'linenos': directives.flag,
|
|
}
|
|
|
|
def run(self):
|
|
config = AutoRun.config
|
|
language = self.arguments[0]
|
|
|
|
if language not in config:
|
|
raise RunBlockError('Unknown language %s' % language)
|
|
|
|
# Get configuration values for the language
|
|
args = config[language].split()
|
|
input_encoding = config.get(language+'_input_encoding', 'utf8')
|
|
output_encoding = config.get(language+'_output_encoding', 'utf8')
|
|
prefix_chars = config.get(language+'_prefix_chars', 0)
|
|
show_source = config.get(language+'_show_source', True)
|
|
|
|
# Build the code text
|
|
proc = Popen(args, bufsize=1, stdin=PIPE, stdout=PIPE, stderr=PIPE)
|
|
codelines = (line[prefix_chars:] for line in self.content)
|
|
code = u'\n'.join(codelines).encode(input_encoding)
|
|
|
|
# Run the code
|
|
stdout, stderr = proc.communicate(code)
|
|
|
|
# Process output
|
|
if stdout:
|
|
out = stdout.decode(output_encoding)
|
|
if stderr:
|
|
out = stderr.decode(output_encoding)
|
|
|
|
# Get the original code with prefixes
|
|
if show_source:
|
|
code = u'\n'.join(self.content)
|
|
code_out = u'\n'.join((code, out))
|
|
else:
|
|
code_out = out
|
|
|
|
literal = nodes.literal_block(code_out, code_out)
|
|
literal['language'] = language
|
|
literal['linenos'] = 'linenos' in self.options
|
|
return [literal]
|
|
|
|
|
|
def setup(app):
|
|
app.add_directive('runblock', RunBlock)
|
|
app.connect('builder-inited', AutoRun.builder_init)
|
|
app.add_config_value('autorun_languages', AutoRun.config, 'env')
|