import paramiko
import time
import sys

def copyfinxos(verno, device):
    ssh=paramiko.SSHClient()
	ssh.load_system_host_keys()
	ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
	ssh.connect('10.1.1.50',port=22,username='admin',password='cisco',compress=True)

	if device == 'A':
		cmd = ('connect nxos a\n').encode()
	elif device == 'B':
		cmd = ('connect nxos b\n').encode()
	else:
		print('device is error')

	chan=ssh.invoke_shell()
	time.sleep(1)
	x = chan.recv(2048).decode()

	chan.send(cmd)

	time.sleep(1)
	x = chan.recv(2048).decode()

	chan.send(('show run' + r' | ' + 'no-more\n').encode())

	time.sleep(5)
	x = chan.recv(409600).decode()

	configfiletmp = '/python/DCfile/' + verno + '/' + 'FI-' + device + '.txt'
	configtmp = open(configfiletmp, 'w')
	configtmp.write(x)
	configtmp.close()
	chan.close()
	ssh.close()

if __name__ == '__main__':
	copyfinxos('v2', 'A')