From b5fbc23f746c86cc985711026d6585991c3251f4 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sat, 28 Apr 2018 22:14:23 -0400 Subject: fixes --- inkstitch.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) (limited to 'inkstitch.py') diff --git a/inkstitch.py b/inkstitch.py index bd3f3f48..fe8d6ecb 100644 --- a/inkstitch.py +++ b/inkstitch.py @@ -5,15 +5,11 @@ from inkstitch.utils import save_stderr, restore_stderr from inkstitch import extensions -def get_extension(): - parser = ArgumentParser() - parser.add_argument("--extension") - args, extras = parser.parse_known_args() +parser = ArgumentParser() +parser.add_argument("--extension") +my_args, remaining_args = parser.parse_known_args() - return args.extension - - -extension_name = get_extension() +extension_name = my_args.extension extension_class = getattr(extensions, extension_name.capitalize()) extension = extension_class() @@ -21,7 +17,7 @@ exception = None save_stderr() try: - extension.affect() + extension.affect(args=remaining_args) except (SystemExit, KeyboardInterrupt): raise except Exception: -- cgit v1.2.3