PrettyTable_Color_Asyncio_Netdev.py

#!/usr/bin/python3

import asyncio
import netdev
import os, time, sys ,socket, json

from termcolor import colored, cprint
from prettytable import PrettyTable

USER = 'cisco'
PASSWORD = 'Cisco123'

start = time.time()

# Define username and password to login to all routers with

Switches_dic=[]
Switches_IP=[]

#colored('Device Name' , 'red','on_white')

Table = PrettyTable([colored('Device Name' , 'magenta', attrs=['bold']), "IP Address", "Status"])


with open('IPaddress_Switches.txt') as f:
    for line in f:
        line = line.strip()
        try:
            socket.inet_aton(line)
            Switches_IP.append(line)
            IP = { 'username' : USER,'password' : PASSWORD,'device_type': 'cisco_ios','host': line}
            Switches_dic.append(IP)

        except socket.error:
            print ("Invalid IP address  " + line)

cprint ("This is Switches IPs: \n","red")
print (Switches_IP)
print("\n")

os.chdir("output/")

async def task(param):

    async with netdev.create(**param) as ios:
        # Get the device Hostname
        hostname = await ios.send_command("show run | inc hostname")
        hostname = hostname[9:]
        hostname.split(" ")

        print (hostname, end='') ,print(colored("  Done","yellow"))

        Table.add_row([hostname,param['host'], "Done"])

        output = await ios.send_command("show run")
 
        f = open((hostname), "w")
        print((output), file=f)  # python 3.x



async def run():

    
    tasks = [task(switch) for switch in Switches_dic]
    await asyncio.wait(tasks)


loop = asyncio.get_event_loop()
loop.run_until_complete(run())

end = time.time()
total = int(end - start)

t1= colored('\n Elapsd Time: ' , 'white')
t2=(str(total) + " Sec\n")
t3=colored(t2,'green')

print(t1+t3)

print(Table.get_string(title="APE-360"))

sys.exit(1)