revert to use __main__.py

so that python3 -m iriscasttools <args> can work
anish-mudaraddi · Mar 31, 2023 · 6e0bd30 · 6e0bd30
1 parent b69dd45
commit 6e0bd30
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/iriscasttools/iriscasttools/__init__.py b/iriscasttools/iriscasttools/__init__.py
@@ -0,0 +1 @@
+from iriscasttools import __main__
diff --git a/iriscasttools/iriscasttools/main.py → iriscasttools/iriscasttools/__main__.py b/iriscasttools/iriscasttools/main.py → iriscasttools/iriscasttools/__main__.py
@@ -11,7 +11,11 @@ def _prep_logging():
     logger.addHandler(logging.StreamHandler(sys.stdout))
 
 
-if __name__ == "__main__":
+def main():
     _prep_logging()
     cmd_args = parse_args(sys.argv[1:])
     print(get_iriscast_stats(cmd_args.as_csv, cmd_args.include_header))
+
+
+if __name__ == "__main__":
+    main()