没啥技术含量可言，注意的是要想把配置一口气输完而不用按空格显示下一页，需要执行skip-page-display, cisco的交换机应该是 terminal length 0 。
注意ssh.before和pexpect.EOF
#!/usr/bin/env python
import os, sys,pexpect
''' collect all the configration of actively running switches/firewalls '''
sw_list = ['172.16.1.232', '172.16.1.233', '172.16.1.234', '172.16.1.235', '172.16.1.236', '172.16.1.239', '172.16.1.240', '172.16.1.242', '172.16.1.243', '172.16.1.244', '172.16.1.245', '172.16.1.246', '172.16.2.248' ]
user="xxx"
passwd="xxxx"
cmd="show running-config"
def collect_config(ip) :
try:
ssh = pexpect.spawn('ssh %s@%s ' % (user,ip), timeout=120)
i = ssh.expect(['(?i)password','continue connecting (yes/no)?','[$#>]','No route to host','pexpect.TIMEOUT'])
if i == 0:
ssh.sendline(passwd)
elif i == 1:
ssh.sendline('yes')
ssh.expect('password')
ssh.sendline(passwd)
elif i == 2:
ssh.sendline()
elif i == 3:
print 'couldn\'t connect to host ',ssh.before
else :
print 'ssh to host timeout ,please check network and pasword '
ssh.expect('[$#>]')
if ip == '172.16.2.248' :
#ssh.logfile_read = sys.stdout
ssh.sendline('config system console')
ssh.expect('[>$#]')
# display output without more ..., so that you won't hit <SPACE> to continue next page.
ssh.sendline('set output standard')
ssh.expect('[>$#]')
ssh.sendline('end')
ssh.expect('[>$#]')
ssh.sendline("show full-configuration")
f = open('./' + ip , "w")
ssh.logfile_read = f
ssh.expect('[>$#]')
print "\n #################### %s done ######################### \n"%ip
ssh.sendline('exit')
# without EOF, buffer probably won't flush into file.
ssh.expect(pexpect.EOF)
else :
ssh.sendline('en')
ssh.expect('[>$#]')
ssh.sendline('skip-page-display')
ssh.expect('[>$#]')
ssh.sendline(cmd)
f=open('./' + ip , "w")
ssh.logfile_read = f
ssh.expect('[>$#]')
#print ssh.before # everything in the previous after the previous expect.
print "\n #################### %s done ######################### \n"%ip
ssh.sendline('exit')
ssh.expect('>')
ssh.sendline('exit')
ssh.expect(pexpect.EOF)
# Not quite clear why do we need to close and re-open the file again, but it doesn't work at all if not.
# get rid of useless output as following:
#########################################################
# SSH@ICX6450-48 Switch>exitConnection to 172.16.1.232 closed by remote host.
#########################################################
f.close()
f = open('./' + ip , "r")
while True :
lines = f.readlines()
if lines :
x = open("./%s.conf"%ip ,'w')
if ip == '172.16.2.248' : # The output of Firewall is a little bit different from Switches.
x.writelines(lines[2:-3])
else :
x.writelines(lines[2:-4])
x.close()
else :
break
ssh.close()
f.close()
os.remove(ip)
except Exception,e :
f.close()
ssh.close()
print " connect error,",str(e)
sys.exit(1)
for i in sw_list:
collect_config(i)
或者干脆把机器list和credential放到另外一个模块里面然后
class new:
def __init__(self):
pass
def switches(self):
return {'bj': ['172.16.1.232', '172.16.1.233', '172.16.1.234', '172.16.1.235', '172.16.1.236', '172.16.1.239', '172.16.1.240', '172.16.1.242', '172.16.1.243', '172.16.1.244', '172.16.1.245', '172.16.1.246', '172.16.2.248
' ], 'wuhan':[]}
# call
sw_list = inventory.new().switches()['bj']
