Personal_Python_Clients/main.py at main · miloandrs/Personal_Python_Clients · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#!/usr/bin/env Python3

"""
Napalm Personal Client Tests.
"""

# imports
import device_pooler
import executioner
from colorama import Fore, Back, Style
import re
import constants

# Global declaration


def main():
    """
    Main runtime code.
    """
    while True:
        print(
            """
            Please select your use case
            Q or q to quit

            ***
            '1' : get_device_facts,
            '2' : get_device_firmware_version,
            '3' : get_device_environment,
            '4' : get_device_interfaces,
            '5' : get_device_fatal_errors,
            '6' : get_device_mac_address_table,
            '7' : get_bin_file_present,
            '8' : configure_interface_access,
            '9' : configure_interface_trunk,

            """
        )

        selection = input("Please enter your request: ")
        if re.match("^[qQ]$", selection):
            return True
        elif re.match("^[1-9]$|^[1-2][0]$|^(10)$", selection):
            devices = device_pooler.site_devices_selection(constants.site)
            for device in devices:
                try:
                    val = executioner.devices_interaction(device, selection)
                    print(val)
                except Exception:
                    print(Fore.RED + Style.BRIGHT + "Unable to connect, bad credentials or device issues" + Style.RESET_ALL)
                    continue
        else :
            print(Fore.RED + Style.BRIGHT + "Please enter a valid choice!" + Style.RESET_ALL)
            continue


if __name__ == "__main__":
    main()