Add a __main__.py module with main()

lpozo · lpozo · commit 693616c87c9c · 2021-03-01T20:40:55.000-05:00
diff --git a/rptree/__main__.py b/rptree/__main__.py
@@ -0,0 +1,23 @@
+"""This module provides the RP Tree CLI."""
+
+import pathlib
+import sys
+
+from .cli import parse_cmd_line_arguments
+from .rptree import DirectoryTree
+
+
+def main():
+    args = parse_cmd_line_arguments()
+    root_dir = pathlib.Path(args.root_dir)
+    if not root_dir.is_dir():
+        print("The specified root directory doesn't exist")
+        sys.exit()
+    tree = DirectoryTree(
+        root_dir, dir_only=args.dir_only, output_file=args.output_file
+    )
+    tree.generate()
+
+
+if __name__ == "__main__":
+    main()
