From 395067f97e641110727c44bb2605050338fe3b98 Mon Sep 17 00:00:00 2001 From: Lex Neva Date: Sat, 28 Jul 2018 20:40:14 -0400 Subject: support multiword extension names --- inkstitch.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'inkstitch.py') diff --git a/inkstitch.py b/inkstitch.py index a9ce829e..2e21d964 100644 --- a/inkstitch.py +++ b/inkstitch.py @@ -10,7 +10,11 @@ parser.add_argument("--extension") my_args, remaining_args = parser.parse_known_args() extension_name = my_args.extension -extension_class = getattr(extensions, extension_name.capitalize()) + +# example: foo_bar_baz -> FooBarBaz +extension_class_name = extension_name.title().replace("_", "") + +extension_class = getattr(extensions, extension_class_name) extension = extension_class() exception = None -- cgit v1.2.3