aboutsummaryrefslogtreecommitdiffstats
path: root/main.py
blob: ca23f58771aa62d0d257fdf67f27300a9e3fca75 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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__":
    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.")