之前发布了一篇关于网络设备自动化巡检脚本,功能已基本满足日常的需求,但是,随着网络设备的数量不断增加,巡检的时间耗时也比较长。为了解决这个脚本巡检耗时长的问题,对该脚本做了一些优化。
简单回顾一下脚本的使用方法
脚本是使用python3去编写的,主要使用到telnetlib和paramiko。在使用之前先要安装这两个插件库,具体的安装如下(前提环境中已经安装python3)
pip install paramiko telnetlib
脚本中定义了三个类,一个是基类,另外两个类继承基类,分别实现ssh方法和telnet方法。
使用方法也很简单,在if __name__ == '__main__'代码块下,定义要巡检的网络设备,如下图
SSHtool(host=ip,username='admin',password='admin')这是实例化对象,实例化需要传入参数,最少要传入三个参数,host,username,password这三个是必须的,如果网络设备需要super密码,直接在后面添加就行啦。例如:
ssh_obj = SSHtool(host=ip,username='admin',password='admin',superpassword='admin')
ssh_obj.ssh_on(commands)是调用ssh的方法执行巡检,传入要执行命令
ssh_obj.savefile()是调用savefile()方法,把巡检结果保存到文件
执行演示
从上图看到,执行两台设备,两台命令耗时36.7s。如果设备逐渐增多,那么巡检的时间耗时就会变长。那么如何能尽可能地把耗时缩短呢。
引入多线程
刚才的脚本是在单线程运行的,意思是,需要等第一台设备运行完之后,才会执行第二台设备。这样显然不合理。下面把这个脚本稍微的修改一下。使用多线程需要导入多线程的扩展包。
import threading
导入扩展包后,需要对之前运行脚本进行修改,具体修改如下
if __name__ == '__main__': commands = ['display ip interface brief', 'display cpu-usage'] ip_list = ['192.168.56.10', '192.168.56.20'] for ip in ip_list: ssh_obj = SSHtool(host=ip, username='admin', password='admin') t = threading.Thread(target=ssh_obj.ssh_on, args=(commands,)) t.start()
通过多线程,巡检时间大大缩短了一半,从原来的36.7秒,缩短到18.1秒。但设备越多就越明显的。