'''

Description: Takes a file of IP addresses as first argument, command as second argument
and runs the command on all the devices

Example: python3 paramiko-test.py devices "show ver | inc uptime"

'''

from paramiko import client
import argparse

parser = argparse.ArgumentParser(description='Send command to cisco device.')
parser.add_argument('infile', metavar='input file', type=str, nargs=1, help='list of device ip addresses')
parser.add_argument('command', metavar='command', type=str, nargs=1, help='send command to Cisco device')

args = parser.parse_args()
infile = args.infile[0]
command = args.command[0]


class ssh:
    client = None

    def __init__(self, address, username, password):
        print("Connecting to server.")
        self.client = client.SSHClient()
        self.client.set_missing_host_key_policy(client.AutoAddPolicy())
        self.client.connect(address, username=username, password=password, look_for_keys = False)

    def sendCommand(self, command):
        if(self.client):
            stdin, stdout, stderr = self.client.exec_command(command)
            while not stdout.channel.exit_status_ready():
                if stdout.channel.recv_ready():
                    alldata = stdout.channel.recv(1024)
                    while stdout.channel.recv_ready():
                        alldata += stdout.channel.recv(1024)

                    print(str(alldata, "utf8"))

        else:
            print("Connection not opened")

with open(infile) as f:
    address = f.readlines()
    address = [x.strip() for x in address]
    for node in address:
        if (node != ""):
            connection = ssh(node, "cisco", "cisco")
            connection.sendCommand(command)