From 98e35a3fa02b598d104cde9aafb877006d8065b3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?H=C3=A5kon=20Solbj=C3=B8rg?= <hlsolbjorg@gmail.com>
Date: Mon, 1 Apr 2019 22:43:27 +0200
Subject: =?UTF-8?q?feat:=20Add=20argparse=20to=20allow=20for=20specifying?=
 =?UTF-8?q?=20which=20labler=20function=20to=20run=20=F0=9F=A4=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

(limited to 'main.py')

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.")
-- 
cgit v1.2.3