之前发布了一篇关于网络设备自动化巡检脚本,功能已基本满足日常的需求,但是,随着网络设备的数量不断增加,巡检的时间耗时也比较长。为了解决这个脚本巡检耗时长的问题,对该脚本做了一些优化。

简单回顾一下脚本的使用方法

脚本是使用python3去编写的,主要使用到telnetlib和paramiko。在使用之前先要安装这两个插件库,具体的安装如下(前提环境中已经安装python3)

pip install paramiko  telnetlib

脚本中定义了三个类,一个是基类,另外两个类继承基类,分别实现ssh方法和telnet方法。




python批量巡检操作系统源码 python自动化巡检脚本_扩展包


使用方法也很简单,在if __name__ == '__main__'代码块下,定义要巡检的网络设备,如下图


python批量巡检操作系统源码 python自动化巡检脚本_如何导入全部obj_02


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()方法,把巡检结果保存到文件

执行演示


python批量巡检操作系统源码 python自动化巡检脚本_python批量巡检操作系统源码_03


从上图看到,执行两台设备,两台命令耗时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秒。但设备越多就越明显的。


python批量巡检操作系统源码 python自动化巡检脚本_网络设备_04