aboutsummaryrefslogtreecommitdiffstats
path: root/cables.py
diff options
context:
space:
mode:
Diffstat (limited to 'cables.py')
-rw-r--r--cables.py87
1 files changed, 0 insertions, 87 deletions
diff --git a/cables.py b/cables.py
deleted file mode 100644
index a7ff5de..0000000
--- a/cables.py
+++ /dev/null
@@ -1,87 +0,0 @@
-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)