1 files changed, 60 insertions, 0 deletions
diff --git a/tech-support/labels/main.py b/tech-support/labels/main.py
new file mode 100644
index 0000000..b4f7c53
--- /dev/null
+++ b/tech-support/labels/main.py
@@ -0,0 +1,60 @@
+import argparse
+import sys
+
+from cables import make_cable_labels
+from gondul import fetch_gondul_switches
+from switches import make_switch_labels
+
+parser = argparse.ArgumentParser(
+    "Label generator script 2000",
+    formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+parser.add_argument("labler", type=str,
+                    help="The label function to run. Either [c]ables or [s]witches.")
+parser.add_argument("--gondul-user", type=str,
+                    help="Gondul username. Overrides env GONDUL_USERNAME")
+parser.add_argument("--gondul-pass", type=str,
+                    help="Gondul password. Overrides env GONDUL_PASSWORD")
+parser.add_argument("--gondul-api", type=str,
+                    help="Gondul API base. Overrides env GONDUL_API")
+parser.add_argument("--gondul-switches", type=str,
+                    help="Gondul switches endpoint. Overrides env GONDUL_SWITCHES_ENDPOINT")
+parser.add_argument("--match-switches", type=str, default="^e(.*)",
+                    help="Regex for matching switches")
+parser.add_argument("--outfile", "-o", type=str, default="cable_labels.csv",
+                    help="Output (base) file name. Might be appended with numbers for cables.")
+
+cables_args = parser.add_argument_group("cables")
+cables_args.add_argument("--ap", type=str, action="append",
+                         help="Name of a switch where an AP should be connected")
+cables_args.add_argument("--aps-file", type=str,
+                         help="Path to a newline-separated file with switches where an AP should be connected")
+cables_args.add_argument("--copies", "-c", type=int, default=2,
+                         help="Number of copies per label")
+cables_args.add_argument("--uplinks", "-u", type=int, default=3,
+                         help="Number of uplinks per switch")
+cables_args.add_argument("--split", "-s", type=int, default=100,
+                         help="Split into CSV files of this size")
+
+if __name__ == "__main__":
+    args = parser.parse_args()
+
+    switches = fetch_gondul_switches(
+        api=args.gondul_api,
+        endpoint=args.gondul_switches,
+        username=args.gondul_user,
+        password=args.gondul_pass,
+        match=args.match_switches,
+    )
+
+    if args.labler[0] == "c":
+        make_cable_labels(switches,
+                          aps=args.ap,
+                          ap_file=args.aps_file,
+                          copies=args.copies,
+                          outfile=args.outfile,
+                          split_per_num=args.split)
+    elif args.labler[0] == "s":
+        make_switch_labels(switches, outfile=args.outfile)
+    else:
+        parser.print_help()
+        sys.exit("Invalid labler operation.")