Python脚本巡检华为交换机——筑梦之路

# -*- coding: utf-8 -*-
from netmiko.huawei.huawei import HuaweiSSH
from netmiko import NetMikoTimeoutException
from netmiko import NetMikoAuthenticationException
from getpass import getpass
import re
import io
import xlwt


def main():
    """
        主函数
    """
    # 让用户输入ssh用户名密码
    username = input('请输入ssh用户名：')
    password = getpass('请输入ssh密码：')

    # 打开ip_list.txt文件获取IP列表
    ip_list = open('ip_list.txt', 'r')
    ip_addr = ip_list.readlines()
    ip_list.close()

    cmd_line = ['display version', 'display cpu-usage', 'display memory-usage', 'display device']
 
    # 创建一个workbook 设置编码
    workbook = xlwt.Workbook(encoding='utf-8')
    # 创建一个worksheet
    worksheet = workbook.add_sheet('My Worksheet')
    # 初始化表格
    worksheet.write(0, 0, label = "交换机IP")
    worksheet.write(0, 1, label = "交换机名称")
    worksheet.write(0, 2, label = "软件版本")
    worksheet.write(0, 3, label = "CPU利用率")
    worksheet.write(0, 4, label = "内存利用率")
    worksheet.write(0, 5, label = "硬件状态")
    hang = 0
    lie = 0

    # 遍历ip列表用来生成迭代器
    for ip in iter(ip_addr):
        print(' ')
        print('本次巡检的设备IP：' + ip)
        try:
            S5720 = {
                'device_type': 'huawei',
                'ip': ip,
                'username': username,
                'password': password,
            }
            # 实例化HuaweiSSH
            net_connect = HuaweiSSH(**S5720)
            #print ("恭喜，成功登录")
            #print ("设备名：" + str(net_connect.find_prompt().strip('<>')))
            ip_str = (ip)
            hang = hang + 1
            # 初始化表格列
            lie = 0
            worksheet.write(hang, lie, label=ip_str)
            lie = lie + 1
            worksheet.write(hang,
                            lie,
                            label=net_connect.find_prompt().strip('<>'))
            for cmd in iter(cmd_line):
                cmd_result = net_connect.send_command(cmd)
                regex_str = []
                if 'VRP (R) software' in cmd_result:
                    regex_str = '\(\w*\d\d.*\)'
                    version = (re.search(regex_str, cmd_result))
                    lie = lie + 1
                    worksheet.write(hang,
                                    lie,
                                    label=version.group().strip('()'))
                    cmd_result = ''
                if 'CPU ' in cmd_result:
                    regex_str = '\d*.\d*.\%'
                    cpu_usage = (re.search(regex_str, cmd_result))
                    lie = lie + 1
                    worksheet.write(hang,
                                    lie,
                                    label=cpu_usage.group().strip(' '))
                    cmd_result = ''
                if 'Memory ' in cmd_result:
                    regex_str = '\d*.\%'
                    memory = (re.search(regex_str, cmd_result))
                    lie = lie + 1
                    worksheet.write(hang, lie, label=memory.group())
                    cmd_result = ''
                if 'Device ' in cmd_result:
                    if 'Abnormal' in cmd_result:
                        lie = lie + 1
                        worksheet.write(hang, lie, label=u"Abnormal")
                    elif 'WrongType' in cmd_result:
                        lie = lie + 1
                        worksheet.write(hang, lie, label=u"WrongType")
                    elif 'Unregistered' in cmd_result:
                        lie = lie + 1
                        worksheet.write(hang, lie, label=u"Unregistered")
                    elif 'Off' in cmd_result:
                        lie = lie + 1
                        worksheet.write(hang, lie, label=u"Off")
                    elif 'Offline' in cmd_result:
                        lie = lie + 1
                        worksheet.write(hang, lie, label=u"Offline")
                    else:
                        lie = lie + 1
                        worksheet.write(hang, lie, label=u"Normal")
                    cmd_result = ''

            net_connect.disconnect()
        except (EOFError, NetMikoTimeoutException):
            print('无法连接设备')
            netmikotimeout = (u'无法连接设备' + ip)
            hang = hang + 1
            lie = 0
            worksheet.write(hang, lie, label=netmikotimeout)
        except (EOFError, NetMikoAuthenticationException):
            print('用户名密码错误!')
            netmikotuehenticattion = (u'用户名密码错误' + ip)
            hang = hang + 1
            lie = 0
            worksheet.write(hang, lie, label=netmikotimeout)
        workbook.save('xunjian.xls')


if __name__ == '__main__':
    main()

pip install netmiko
pip install xlwt

创建一个名称为：ip_list.txt 的文件用来存放巡检的设备IP