#!/usr/bin/env python
# -*- coding: utf-8 -*-
#python2.7脚本 
# 华三系列交换机可以备份

username='admin'
password='admin'
StartFtpPath="./txt/startHuasanFtp"  #开启FTP命令的位置
StopFtpPath="./txt/stopHuasanFtp"	  #关闭FTP命令的位置
switchpath="./txt/test.txt"	        #交换机IP地址所在的位置

#连接到telnet服务器并执行开启ftp服务的命令
def do_telnet1(swip,username,password,finish):
	import telnetlib,time
	#连接到telnet服务器
	tn=telnetlib.Telnet(swip)
	tn.set_debuglevel(2)
	#输入用户名
	tn.read_until('Username:')
	tn.write(username + '\n')
	#输入密码
	tn.read_until('Password:')
	tn.write(password + '\n')
	#执行命令
	tn.read_until(finish)
	for cmd in open(StartFtpPath):
		tn.write('%s' % cmd)
	#关闭telnet连接
	time.sleep(1)
	tn.set_debuglevel(0)
	tn.close()

#连接到telnet服务器并执行关闭ftp服务的命令
def do_telnet2(swip,username,password,finish):
	import telnetlib,time
	#连接到telnet服务器
	tn=telnetlib.Telnet(swip)
	tn.set_debuglevel(5)
	#输入用户名
	tn.read_until('Username:')
	tn.write(username + '\n')
	#输入密码
	tn.read_until('Password:')
	tn.write(password + '\n')
	#执行命令
	tn.read_until(finish)
	for cmd in open(StopFtpPath):
		tn.write('%s' % cmd)
	#关闭telnet连接
	time.sleep(1)
	tn.set_debuglevel(0)
	tn.close()

#ftp备份
def ftp_bak(name1,name2,swip):
	import ftplib,os
	ftp=ftplib.FTP()
	ftp.connect(swip,21,10)
	ftp.login(username,password)
	print ftp.getwelcome()
	ftp.cwd('/')
	try:
		fp=open('./bak/'+name1,'wb')
		ftp.retrbinary('RETR vrpcfg.cfg',fp.write,1024)
	except ftplib.error_perm,e:
		fp.close()
		os.remove('./bak/'+name1)
		fp=open('./bak/'+name2,'wb')
		ftp.retrbinary('RETR config.cfg',fp.write,1024)
	fp.close()
	print 'OK!'
	
#开启交换机的FTP服务，并备份配置
def start_ftp():
	finish='>'
	for swip1 in open(switchpath):
		swip=swip1.strip()
		import time
		thedate=time.strftime('%Y-%m-%d',time.localtime())
		name1=swip+'_'+'vrpcfg'+thedate+'.cfg'
		name2=swip+'_'+'config'+thedate+'.cfg'
		do_telnet1(swip,username,password,finish)
		ftp_bak(name1,name2,swip)

#关闭交换机的FTP服务
def stop_ftp():
	finish='>'
	for swip1 in open(switchpath):
		swip=swip1.strip()
		do_telnet2(swip,username,password,finish)
	print 'OK!'
		
#开始执行		
start_ftp()
stop_ftp()