diff options
-rw-r--r-- | support/asymptote/__main__.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/support/asymptote/__main__.py b/support/asymptote/__main__.py index 1816224..5c6f67a 100644 --- a/support/asymptote/__main__.py +++ b/support/asymptote/__main__.py @@ -1,16 +1,23 @@ -import os, sys +import os, sys, shutil from lib import util -def _asymptote(in_path, out_path, asymptote_dir): - util.command([os.environ['ASYMPTOTE'], '-f', 'pdf', '-o', out_path, in_path], set_env = { 'ASYMPTOTE_DIR': asymptote_dir }) +def _asymptote(in_path, out_path, asymptote_dir, working_dir): + util.command([os.environ['ASYMPTOTE'], '-f', 'pdf', '-o', out_path, in_path], set_env = { 'ASYMPTOTE_DIR': asymptote_dir }, working_dir = working_dir) def main(in_path, out_path): _, out_suffix = os.path.splitext(out_path) if out_suffix == '.pdf': - _asymptote(in_path, out_path, os.path.dirname(in_path)) + with util.TemporaryDirectory() as temp_dir: + absolute_in_path = os.path.abspath(in_path) + temp_out_path = os.path.join(temp_dir, 'out.pdf') + + # Asymptote creates A LOT of temp files (presumably when invoking LaTeX) and leaves some of them behind. Thus we run asymptote in a temporary directory. + _asymptote(absolute_in_path, 'out', os.path.dirname(absolute_in_path), temp_dir) + + shutil.copyfile(temp_out_path, out_path) else: raise Exception('Unknown file type: {}'.format(out_suffix)) |