python批量拷贝数据脚本_使用python来玩转ensp~3-写个批量备份配置文件脚本

目录回顾：

根据前面所学习的知识，可以远程登录设备下发命令等，现在已经可以转化为生产力了哈，今天小Q要实现的就是一个批量备份配置的小东西，

小Q通过学习产品文档了解到，可以通过以下四种方法进行备份配置文件：

a)直接屏幕拷贝。

b)备份配置文件到存储器中。

c)通过TFTP备份配置文件。

d)通过FTP备份配置文件。

小Q对这几种备份方式做了一个小对比，

a)直接屏幕拷贝 -屏幕上显示的配置信息受终端软件的影响，可能会出现某配置过长而换行的情况，而换行的配置可能无法恢复。 -丢弃此方式

b)备份配置文件到存储器中 ?受ENSP环境所限制，可以实现么？ 不了解，丢弃此方式

c)通过TFTP备份配置文件 ?对设备无修改无变动 ?完美啊

d)通过FTP备份配置文件 ?备份时需要启动FTP服务器功能，对设备配置有变动，涉及安全性-丢弃此方式

既然采用TFTP的方式备份配置文件，

采用如下的思路进行配置：

a)保存配置文件。-执行命令 save config.cfg

b)通过TFTP备份配置文件。-需要一个tftp server,然后在设备上执行上传命令

组网图如下

要完成的备份脚本程序的组成模块如下：

首先实现备份程序的功能，直接修改第2章的内容，将多个设备信息已固定格式保存在一个txt里面，如下：

IP:用户名:密码

修改上次的ssh登录，使其支持多设备，代码如下(代码中依然增加了注释)：

#!/usr/bin/python

#coding=utf-8

# 批量备份配置文件小脚本

from Exscript.util.interact import read_login

from Exscript.protocols import SSH2

from Exscript import Account

import os

import time

import datetime

import sys

reload(sys)

sys.setdefaultencoding("utf-8")

def LoginSSH(_Host,_Username,_Password) :

print "[+] Loging Host : %s " % _Host

account = Account(_Username,_Password)

ssh = SSH2()

try :

print "[+] Trying to Login in with username: %s password: %s " % (_Username,_Password)

ssh.connect(_Host)

ssh.login(account)

except Exception,e:

print "[-] Failed! ...",e

return 'LoginError',ssh

print "[+] Success ... username: %s and passoword %s is VALID! " % (_Username, _Password)

return 'LoginSuccess',ssh

def StartExecuteBackup(sshConnection,SaveCfgFileName):

BackUpCommand = 'save %s' % SaveCfgFileName

sshConnection.execute('dir')

# 判断是否已经存在备份文件，发现=1，未发现=0

CfgFileStatus = 0

if SaveCfgFileName in sshConnection.response:

CfgFileStatus = 1

sshConnection.set_prompt(r'y/n')

sshConnection.execute(BackUpCommand)

# 备份文件不存在时，确认一次

if CfgFileStatus == 0:

sshConnection.set_prompt()

sshConnection.execute('y')

print '[*] Save the configuration to %s' % SaveCfgFileName

# 备份文件存在时，确认两次，进行覆盖

else:

sshConnection.execute('y')

sshConnection.set_prompt()

sshConnection.execute('y')

print '[*] Overwrite %s '% SaveCfgFileName

def StartExecuteTftpUpload(sshConnection,TFtpHost,SaveCfgFileName):

# tftp IP put config.cfg backup.cfg

print '[*] Uploading the file to the remote TFTP server. Please wait...'

TFtpUpFileCommand = "tftp %s put %s %s" % (TFtpHost,SaveCfgFileName,SaveCfgFileName)

sshConnection.execute(TFtpUpFileCommand)

print '[*] TFTP: Uploading the file successfully.'

if __name__ == '__main__':

TFtpHost = raw_input("[*] Please Input Tftp server Adress:>")

# 读取多设备列表文件

HostFile = 'HostList.txt'

Date = datetime.date.today().strftime("%Y-%m-%d")

for line in open(HostFile, "r").readlines():

HostinfoList = line.strip().split(':')

_Host = HostinfoList[0]

_Username = HostinfoList[1]

_Password = HostinfoList[2]

LoginResult,sshConnection = LoginSSH(_Host,_Username,_Password)

# 登录成功后即可执行命令,基本检查信息

if LoginResult == 'LoginSuccess':

# 设置生成的配置文件名

SaveCfgFileName = '%s_%s_config.cfg' % (_Host,Date)

# 生成配置文件

StartExecuteBackup(sshConnection,SaveCfgFileName)

# 备份配置文件到tftp服务器

StartExecuteTftpUpload(sshConnection,TFtpHost,SaveCfgFileName)

# 关闭连接

#sshConnection.close()

print '[*] Session Close.'

然后，我们在编写一个tftp server小脚本(其他类似tftp软件一样可以的)，用来接收从设备上传的配置文件，并保存，代码非常简单如下：

#!/usr/bin/python

#coding=utf-8

import sys

reload(sys)

sys.setdefaultencoding("utf-8")

import tftpy

import ConfigParser

config = ConfigParser.RawConfigParser()

# 如果要设置保存的目录，修改tftpcfg.cfg即可

config.read('tftpcfg.cfg')

directory = config.get('config','directory')

server = tftpy.TftpServer(directory)

print '[*] Tftp: 0.0.0.0'

print '[*] Upfile in:',directory

server.listen('0.0.0.0', 69)

使用时要设置tftp server工作目录，修改tftpcfg.cfg即可，如下：

好的现在一些都准备完毕了，首先启动我们的tftp

server软件，

然后运行我们的批量备份配置程序，运行后先输入Tftp服务器的ip，脚本将会自动进行配置备份，效果如下：

此时在查看tftp的目录，已经获取到了配置文件，如下：

目前这个小脚本异常处理还未考虑，所以仅限测试使用哦，相信简单修改下就可以使用了哈。

提示：可以将脚本生成exe的格式，就可以运行在无python环境的电脑了，附件将会进行打包提供下载，方便各位根据实际环境修改。