银狐NetDevOps-网络运维Python初篇(三)openpyxl读取excel信息,批量抓取设备配置

科技银狐

1、训练场景:读取excel大量设备信息,根据得到的信息进行批量操作

上一个章节我们举了一个简单的案例,利用netmiko抓取单台思科设备的配置,现实情况我们不会对1台设备进行操作,那样的话就不需要自动化了。网上有很多教程,教大家用for循环读取txt文档内容,这种方式也确实可以。

不过有些企业有自己的CMDB系统,支持API调用或者excel导出,所以本章内容就以excel内容为例,总归类似的内容相对较少。

2、实验环境:

操作系统:windows 10 PC机

python版本:python 3.8

网络设备:华为CE 6865

编辑器:vscode(pycharm、sublime均可,推荐vscode)

excel格式:初次使用简单一些,excel中只加入IP地址

[图片上传失败...(image-bed0b-1621266802456)]

3、思路分析

使用python第三方库openpyxl读取设备IP地址,并形成一个IP地址的list,这里需要注意第一行是表头,所以要从第二行开始读取。拿到所有设备IP地址列表后进行for循环对每台设备进行SSH连接(netmiko),然后发送CLI命令,并打印出来。

4、整体代码

#!/usr/bin/env python
#coding: utf-8

from netmiko import ConnectHandler
from openpyxl import Workbook
from openpyxl import load_workbook

def read_device_excel( ):

    ip_list = []

    wb1 = load_workbook('/home/netops/venv/cs_lab.xlsx')
    ws1 = wb1.get_sheet_by_name("Sheet1")

    for cow_num in range(2,ws1.max_row+1):

        ipaddr = ws1["a"+str(cow_num)].value
        ip_list.append(ipaddr)

    return ip_list

def get_config(ip_list):

    for ipaddr in ip_list:

        session = ConnectHandler(device_type="huawei",
                                ip=ipaddr,
                                username="dev_user",
                                password="dev_password",
                                banner_timeout=300)

        print("connecting to "+ ipaddr)
        print ("---- Getting HUAWEI configuration from {}-----------".format(ipaddr))

        config_data = session.send_command('dis ip int brief ')

        print (config_data)
        session.disconnect()

def main():
    ip_list = read_device_excel()
    get_config(ip_list)

#-------------------------------------------------------    
if __name__ == '__main__':
    main()

执行结果:

image

5、代码详解

#!/usr/bin/env python
#coding: utf-8

from netmiko import ConnectHandler
from openpyxl import Workbook
from openpyxl import load_workbook

导入netmiko和openpyxl模块就不多说了,大家都懂。

def read_device_excel( ):

    ip_list = []

    wb1 = load_workbook('/home/netops/venv/cs_lab.xlsx')
    ws1 = wb1.get_sheet_by_name("Sheet1")

1、创建read_device_excel函数,因为我们需要一个IP地址的列表,所以先创建一个空list。

2、调用openpyxl穿件workbook对象,用于创建workbook并复制给wb1,workbook1内的sheet1赋值给ws1(Sheet1也是默认sheet)。

    for cow_num in range(2,ws1.max_row+1):

        ipaddr = ws1["a"+str(cow_num)].value
        ip_list.append(ipaddr)

    return ip_list

3、A列第一行可以用A1表示,A1的value属性就是单元格里的内容,比如A1=IP,A2=10.23.80.93,现在我们想遍历出所有IP地址,就要用for循环遍历A列所有内容,这里要注意,第一行我们是不需要的,所以我们真正要遍历的是2-5行,可以使用ws1.max_row得到最大值,这里max_row =5,range的(2,5)其实=2,3,4,这样最后一行就抓不到了,所以要max_row+1,确保第一行所有IP都能抓到。

4、最后把ip列表retrun出来(传递出来)

def get_config(ip_list):

    for ipaddr in ip_list:

        session = ConnectHandler(device_type="huawei",
                                ip=ipaddr,
                                username="dev_user",
                                password="dev_password",
                                banner_timeout=300)

        print("connecting to "+ ipaddr)
        print ("---- Getting HUAWEI configuration from {}-----------".format(ipaddr))

        config_data = session.send_command('dis ip int brief ')

        print (config_data)
        session.disconnect()

5、创建一个get_config的函数,并把上面的ip_list传递进来进行for循环。

6、针对没一个IP地址进行SSH登录,并用print提示我们正在登录哪个设备。

7、直接使用send_command方法发送CLI命令。

8、打印交换机返回的结果,并关闭session。

def main():
    ip_list = read_device_excel()
    get_config(ip_list)

#-------------------------------------------------------    
if __name__ == '__main__':
    main()

为了代码更加清晰,创建了1个主函数main(),并分别调用读取excel函数和getconfig函数。

*SSH也可以单独写成一个函数,get_config内部调用SSH函数,这种方式更灵活。

你可能感兴趣的:(银狐NetDevOps-网络运维Python初篇(三)openpyxl读取excel信息,批量抓取设备配置)