import threading

import netmiko
import getpass

# global variables
_platform = 'cisco_ios'
_username = raw_input("Username?:")
_password = getpass.getpass(prompt="Password?:")
_port = 4000
_data = open("hosts.txt")


def send_command(command, ip):
    try:
        dev_ssh = netmiko.ConnectHandler(
            device_type=platform,
            ip=ip,
            username=_username,
            password=_password,
            port=_port,
            secret=_password,
            )
        output = dev_ssh.send_command_timing(command).split()
        if 'Version' in output:
            output = dev_ssh.send_command_timing('\r\n').split()

        with open(ip + '_wrmem.txt','w+') as file:
            for line in output:
                file.write(line)

        print('{} -- Complete'.format(ip))

    except:
        continue


def main():
    all_devices = [ip for ip in _data]
    thread_list = [threading.Thread(target=send_command, args=(ip,)) for ip in all_devices]

    for thread in thread_list:
        thread.start()
    for thread in thread_list:
        thread.join()

if __name__ == '__main__':
    main()
    print('Complete')