summaryrefslogtreecommitdiff
path: root/inkstitch.py
diff options
context:
space:
mode:
authorLex Neva <github.com@lexneva.name>2018-08-16 15:38:59 -0400
committerLex Neva <github.com@lexneva.name>2018-08-16 15:38:59 -0400
commitf70fa25a1630b55f9568e7c8214e6d03536f978a (patch)
treee73998ff313e281563595c1d716b9ca73a0ead01 /inkstitch.py
parent89e5e5c9dbc036be1e0c2802e2324216c6a9ff0a (diff)
parent0f882f07eb0fc263fcb8e05c1a212199902863da (diff)
Merge branch 'master' into lexelby-windows-output-fix
Diffstat (limited to 'inkstitch.py')
-rw-r--r--inkstitch.py6
1 files changed, 5 insertions, 1 deletions
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