From 096db19a9a465cc667aa5d0ef92df2eeac5955b2 Mon Sep 17 00:00:00 2001 From: Michael Schwarz Date: Fri, 11 Sep 2015 21:55:58 +0200 Subject: Wrap all Python main functions with decorator. This decorator check if a module was called as the main module and catches exceptions. --- support/asymptote/__main__.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) (limited to 'support/asymptote') diff --git a/support/asymptote/__main__.py b/support/asymptote/__main__.py index 91902dd..238332f 100644 --- a/support/asymptote/__main__.py +++ b/support/asymptote/__main__.py @@ -1,4 +1,4 @@ -import os, sys, shutil +import os, shutil from lib import util @@ -6,6 +6,7 @@ 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) +@util.main def main(in_path, out_path): _, out_suffix = os.path.splitext(out_path) @@ -23,12 +24,3 @@ def main(in_path, out_path): shutil.copyfile(temp_out_path, out_path) else: raise Exception('Unknown file type: {}'.format(out_suffix)) - - -try: - main(*sys.argv[1:]) -except util.UserError as e: - print 'Error:', e - sys.exit(1) -except KeyboardInterrupt: - sys.exit(2) -- cgit