28 lines
901 B
Python
28 lines
901 B
Python
from docutils import nodes
|
|
|
|
from docutils.parsers.rst import Directive
|
|
|
|
from shutil import which
|
|
import shlex
|
|
import subprocess
|
|
import tempfile
|
|
|
|
class D2langDirective(Directive):
|
|
has_content = True
|
|
def run(self):
|
|
# TODO : add tempfile for diag_source
|
|
d2_bin = which('d2')
|
|
outdir = self.state.document.settings.env.app.builder.outdir
|
|
srcdir = self.state.document.settings.env.srcdir
|
|
diag_source = self.content
|
|
with tempfile.NamedTemporaryFile() as fp:
|
|
for line in self.content.splitlines():
|
|
fp.write(bytes(line,'utf-8'))
|
|
fp.write(bytes('\n','utf-8'))
|
|
fp.seek(0)
|
|
cmd_line = '%s %s %s' % (d2_bin, fp.name, srcdir + "/test.svg")
|
|
args = shlex.split(cmd_line)
|
|
subprocess.run(args)
|
|
image_node = nodes.image(uri="/test.svg")
|
|
return [image_node]
|