Turn main cubing code into a function which can be imported (#129)

* turn main cubing code into a function which can be imported * format * Merge branch 'master' into main-cubing-function
scalableminds · Sep 13, 2019 · 3748d8c · 3748d8c
1 parent 54d4f39
commit 3748d8c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/wkcuber/__main__.py b/wkcuber/__main__.py
@@ -36,8 +36,7 @@ def create_parser():
     return parser
 
 
-if __name__ == "__main__":
-    args = create_parser().parse_args()
+def main(args):
     setup_logging(args)
 
     scale = tuple(float(x) for x in args.scale.split(","))
@@ -87,3 +86,8 @@ def create_parser():
         )
 
     refresh_metadata(args.target_path)
+
+
+if __name__ == "__main__":
+    parsed_args = create_parser().parse_args()
+    main(parsed_args)