/
Switch_AutoBackUp.py
78 lines (71 loc) · 2.89 KB
/
Switch_AutoBackUp.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import logging
import telnetlib
import time
import datetime
from multiprocessing import Pool
class TelnetClient():
def __init__(self,):
self.tn = telnetlib.Telnet()
# 此函数实现telnet登录主机
def login_host(self,host_ip,username,password):
try:
# self.tn = telnetlib.Telnet(host_ip,port=23)
self.tn.open(host_ip,port=23)
except:
logging.warning('%s网络连接失败'%host_ip)
return False
# 等待login出现后输入用户名，最多等待10秒
self.tn.read_until(b'Username: ',timeout=10)
self.tn.write(username.encode('ascii') + b'\n')
# 等待Password出现后输入用户名，最多等待10秒
self.tn.read_until(b'Password: ',timeout=10)
self.tn.write(password.encode('ascii') + b'\n')
# 延时两秒再收取返回结果，给服务端足够响应时间
time.sleep(2)
# 获取登录结果
# read_very_eager()获取到的是的是上次获取之后本次获取之前的所有输出
command_result = self.tn.read_very_eager().decode('ascii')
if 'Login failed' not in command_result:
logging.warning('%s登录成功'%host_ip)
return True
else:
logging.warning('%s登录失败，用户名或密码错误'%host_ip)
return False
# 此函数实现执行传过来的命令，并输出其执行结果
def execute_some_command(self,command):
# 执行命令
self.tn.write(command.encode('ascii')+b'\n')
time.sleep(2)
# 获取命令结果
command_result = self.tn.read_very_eager().decode('ascii')
# command_result=self.tn.read_all().decode('ascii')
logging.warning('命令执行结果：\n%s' % command_result)
# 退出telnet
def logout_host(self):
self.tn.write(b"quit\n")
#执行交换机备份
def switchbak(i,u,p,c):
telnet_client = TelnetClient()
if telnet_client.login_host(i,u,p) :
telnet_client.execute_some_command(c)
telnet_client.logout_host()
if __name__ == '__main__':
print('开始多进程并发备份')
start=time.time()
#50为进程数，用户可根据客户端及服务器配置自行调整，注意：过大进程易被安全软件识别为恶意攻击
p=Pool(50)
for ip in open('switchs.txt').readlines():
ip=ip.strip()
#交换机具备backup命令的telnet用户
username = 'admin'
#该用户密码
password = 'admin'
#tftp服务器地址
ftphost ='TFTP IP'
filename = ip.replace('.','.')+ '-'+datetime.date.today().strftime('%Y%m%d')+'.cfg'
command1 = 'backup startup-configuration to ' +(ftphost)+' '+ filename
p.apply_async(switchbak,args=(ip,username,password,command1))
p.close()
p.join()
end=time.time()
print('备份完成，耗时：%0.2f 秒' %(end-start) )