1 files changed, 87 insertions, 0 deletions
diff --git a/tech-support/labels/cables.py b/tech-support/labels/cables.py
new file mode 100644
index 0000000..a7ff5de
--- /dev/null
+++ b/tech-support/labels/cables.py
@@ -0,0 +1,87 @@
+from itertools import chain
+
+cable_label_format = "%(switch_name)s-%(switch_num)s-%(cable_name)s"
+mark_twice = True
+num_tabs = 1
+
+
+def generate_label(switch, cable_name):
+    data = {
+        "switch_name": switch.split("-")[0],
+        "switch_num": switch.split("-")[1],
+        "cable_name": cable_name,
+    }
+    label = cable_label_format % data
+    if not mark_twice:
+        return label
+
+    return "{}{}{}".format(label, "\t" * num_tabs, label)
+
+
+def generate_label_copies(switch, cable_name, copies=2):
+    return [generate_label(switch, cable_name) for _ in range(0, copies)]
+
+
+def generate_labels(switches, aps=[], copies=2, uplinks=3):
+    print("Generating {} copies of each label for {} uplinks for {} switches and {} APs ({} labels)".format(
+        copies, uplinks, len(switches), len(aps), (len(switches) * uplinks + len(aps)) * copies))
+
+    labels = []
+    for i in range(0, len(switches)):
+        switch = switches[i]
+        switch_name = switch[1:]
+        cable_labels = [generate_label_copies(
+            switch_name, uplink + 1, copies=copies) for uplink in range(0, uplinks)]
+
+        # Destructure the list of copies into a flat list
+        labels.extend(chain.from_iterable(cable_labels))
+
+        if switch_name in aps:
+            labels.extend(
+                generate_label_copies(switch_name, "AP", copies=copies))
+
+    return labels
+
+
+def write_to_file(data, outfile="cable_labels.csv", filenum=1):
+    outfile_numbered = outfile.replace(".", "-{}.".format(filenum))
+
+    with open(outfile_numbered, "w") as f:
+        f.writelines("\n".join(data))
+
+
+def chunk_list(li, items):
+    for i in range(0, len(li), items):
+        yield li[i:i+items]
+
+
+def write_csv(data, outfile="cable_labels.csv", split_per_num=100):
+    split_data = list(chunk_list(data, split_per_num))
+
+    for i in range(0, len(split_data)):
+        write_to_file(split_data[i], filenum=i+1)
+
+    print("Wrote cable labels to {} files, starting from {}".format(
+        len(split_data), outfile.replace(".", "-1.")))
+
+
+def read_aps_file(path):
+    aps = []
+    with open(path, "r") as f:
+        aps = [line.replace("\n", "").strip() for line in f.readlines()]
+
+    return aps
+
+
+def make_cable_labels(switches, ap_file=None, aps=[], copies=2, outfile="cable_labels.csv", split_per_num=100, uplinks=3):
+    print("Generating labels for cables")
+
+    list_of_aps = aps
+    if ap_file:
+        list_of_aps.extend(read_aps_file(ap_file))
+
+    if len(list_of_aps):
+        print("Generating labels for {} APs".format(len(list_of_aps)))
+
+    labels = generate_labels(switches, copies=copies, uplinks=uplinks, aps=aps)
+    write_csv(labels, outfile=outfile, split_per_num=split_per_num)