#!/usr/bin/env python
# -*- coding: utf-8 -*-
#python2.7脚本 
# 华为s系列2700、5700可以备份

username='lxht'
password='LXht@)!$'
StartFtpPath="E:/ligen/switchbak/txt/StartFtp.txt"    #开启FTP命令的位置
StopFtpPath="E:/ligen/switchbak/txt/StopFtp.txt"	  #关闭FTP命令的位置
switchpath="E:/ligen/switchbak/txt/huaweiip.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(name,swip):
	import ftplib
	ftp=ftplib.FTP()
	ftp.connect(swip,21,10)
	ftp.login(username,password)
	print ftp.getwelcome()
	fp=open('e:/ligen/bak/'+name,'wb')
	ftp.retrbinary('RETR vrpcfg.zip',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())
		name=swip+'_'+'vrpcfg'+thedate+'.zip'
		do_telnet1(swip,username,password,finish)
		ftp_bak(name,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()