feat: Add argparse to allow for specifying which labler function to run 🤙

1 files changed, 16 insertions, 1 deletions

diff --git a/main.py b/main.py
index 49f717b..ca23f58 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,20 @@
+import argparse
+import sys
+
 from switches import make_switch_labels
+from cables import make_cable_labels
 
+parser = argparse.ArgumentParser("Label generator script 2000")
+parser.add_argument("labler", type=str,
+                    help="The label function to run. Either [c]ables or [s]witches.")
 
 if __name__ == "__main__":
-    make_switch_labels()
+    args = parser.parse_args()
+
+    if args.labler[0] == "c":
+        make_cable_labels()
+    elif args.labler[0] == "s":
+        make_switch_labels()
+    else:
+        parser.print_help()
+        sys.exit("Invalid labler operation.")