SYS_UP.py

from netmiko import ConnectHandler, NetMikoAuthenticationException , NetmikoTimeoutException, NetmikoBaseException
from prettytable import PrettyTable
import yaml



class Inventory_db:

    def __init__(self):

        self.hosts_IOS = {}
        self.hosts_IOS_XR ={}
        self.hosts_NOKIA_SROS= {}
        external_data = self.get_external_data()
        self.username = external_data["user_pass"]["my_username"]
        self.my_pass  = external_data["user_pass"]["my_password"]
        self.hosts_IOS["IOS"]  = external_data["IOS_IOS_XE"]
        self.hosts_IOS_XR["IOS_XR"]    = external_data["IOS_XR"]
        self.hosts_NOKIA_SROS["NOKIA_SROS"]  = external_data["NOKIA_SROS"]


    def get_external_data(self):
        host_info = "host_info.yaml"
        with open ("host_info.yaml") as file:
            data = yaml.safe_load(file)
        return data





class Sys_up:

    def __init__(self):
        self.inventory = Inventory_db()
    


    def host_connect_parameters(self, hostname, os_type):
        dev_params = {
            "device_type": os_type,
            "host": hostname,
            "username": self.inventory.username,
            "password": self.inventory.my_pass
            }
        return dev_params




    def connect_to_dev(self, hosts, comand, dev_os):
        hostinf={}
        host_sys_info = []
        for host in hosts:
            print("Gathering data on {}".format(host))
            host_params = self.host_connect_parameters(host, dev_os)
            try:
                with ConnectHandler(**host_params) as dev_connect:
                    result = dev_connect.send_command(comand, use_textfsm=True)
                    #print(result)
                    host_sys_info.append(result[0])
            except NetMikoAuthenticationException:
                print("WARNING!!! AUTHENTICATION FAILED - DEVICE: {} ".format(host))
            except NetmikoTimeoutException:
                print("WARNING!!! DEVICE {} NOT REACHABLE OR SSH NOT ENABLED".format(host))
        return host_sys_info


# for devices with  non use_textfsm support  
    def tl_dev_output(self, hosts, sros_command, dev_os, key_comand_ref="result"):
        per_dev_output_collectionnfo = []
        try:
            for host in hosts:
                output_collection = {}
                print("Gathering data on {}".format(host))
                dev = self.host_connect_parameters(host, dev_os)
                with ConnectHandler(**dev) as dev_connect:
                    dev_output = dev_connect.send_command(sros_command)
                    output_collection["hostname"] = host
                    output_collection[key_comand_ref] = dev_output.split(" : ")[1].strip("sec")
                    per_dev_output_collectionnfo.append(output_collection)
        except NetMikoAuthenticationException:
            print("WARNING!!! AUTHENTICATION FAILED - DEVICE: {} ".format(host))
        except NetmikoTimeoutException:
            print("WARNING!!! DEVICE {} NOT REACHABLE OR SSH NOT ENABLED".format(host))      
        return per_dev_output_collectionnfo





    def global_dev_info(self, hosts, comand, sros_command="", key_comand_ref="result"):
        total_dev_groups = len(hosts)
        host_details = []
        for x_element in range(total_dev_groups):
            for k , v in hosts[x_element].items():
                if k == "NOKIA_SROS":
                    if v == None or v == False:
                        continue
                    else:
                        dev_os = "nokia_srl"
                        _NOKIA_conect_details = self.tl_dev_output(v, sros_command,dev_os, key_comand_ref=key_comand_ref )
                        host_details += _NOKIA_conect_details 
                else:
                    if v == None or v == False:
                        continue
                    else:
                        dev_os = "cisco_ios"
                        host_details += self.connect_to_dev(v,comand, dev_os)
        return host_details



    def sys_up_table(self, devs_grp):
            table_view = PrettyTable(["DEVICES", "UPTIME"])
            devs_info = self.global_dev_info(devs_grp, "show version", sros_command="show uptime", key_comand_ref="uptime")
            if len(devs_info) > 0:
                print("")
                print("        UPTIME PER DEVICE - SUMMARY VIEW  ")
                for info in devs_info:
                    hostname = info["hostname"]
                    uptime = info["uptime"]
                    table_view.add_row([hostname, uptime])
                return table_view
            else:
                print("Device info not found, please check ssh and devices reachability")


app = Sys_up()
sys_up_view = app.sys_up_table([app.inventory.hosts_IOS_XR, app.inventory.hosts_IOS, app.inventory.hosts_NOKIA_SROS])
print(sys_up_view, "\n")



print("MIT License")
print("Copyright (c) 2023 ThierryLab.com")