NetDevOps-通过SNMP获取接口信息

在网络自动化过程中, 通过SSH或netconf或restconf获取设备信息时, 因为不同厂商标准不一样, 导致需要为每个厂商的网络设备都编写对应的脚本, 非常麻烦. 这时SNMP就会显的方便很多, 因为SNMP是一个标准协议, 所有厂家都会遵循, 当然SNMP也有私有OID导致每个厂家获取一些私有属性时也需要编写不同的脚本, 但有些标准协议里面定义好的OID是可以通用的, 所以当需要获取公有OID里面定义的属性时, 使用SNMP就非常方便, 并且编写一个脚本后所有厂家都通用, 比如下面就是一个通过SNMP获取网络设备接口信息的示例, 这个脚本可以适配所有厂商的网络设备, 是不是非常方便?
华为官方MIB库链接: https://info.support.huawei.com/info-finder/tool/zh/enterprise/mib

一、编写一个公共文件common.py, 方便其他模块调用

# common.py
from pysnmp.hlapi import *


def snmp_get(ip, community, oid, port=161, lookup_mib=True, timeout=1, retries=5):
    """
    功能: snmp_get, oid结尾必须是.0
    参数:
     - ip: 设备ip
     - community: 团体属性
     - oid: oid
     - port: 端口
     - lookup_mib: False: 不尝试查询MIB库将返回值的oid解析成字符串, True: 尝试查询MIB库将返回值的oid解析成字符串
     - retries: 重试次数
     - timeout: 第一次查询和每次重试的超时时间, 单位为秒.
    返回值:
     - {'SNMPv2-MIB::sysName.0': 'HUAWEI_CE12800'}
    """
    g = getCmd(SnmpEngine(), CommunityData(community), UdpTransportTarget((ip, port), timeout=timeout, retries=retries),
               ContextData(), ObjectType(ObjectIdentity(oid)), lookupMib=lookup_mib)

    errorIndication, errorStatus, errorIndex, varBinds = next(g)
    if errorIndication or errorStatus or errorIndex:
        return
    return dict([[y.prettyPrint() for x in varBinds for y in x]])


def snmp_walk(ip, community, oid, port=161, lookup_mib=False, timeout=1, retries=5):
    """
    功能: snmp_walk
    参数:
     - ip: 设备ip
     - community: 团体属性
     - oid: oid
     - port: 端口
     - lookup_mib: False: 不尝试查询MIB库将返回值的oid解析成字符串, True: 尝试查询MIB库将返回值的oid解析成字符串
     - retries: 重试次数
     - timeout: 第一次查询和每次重试的超时时间, 单位为秒.
    返回值:
     - {'1.3.6.1.2.1.31.1.1.1.1.2': 'NULL0', '1.3.6.1.2.1.31.1.1.1.1.3': 'InLoopBack0'}
    """
    res = []
    next_cmd = nextCmd(SnmpEngine(), CommunityData(community),
                       UdpTransportTarget((ip, port), timeout=timeout, retries=retries),
                       ContextData(), ObjectType(ObjectIdentity(oid)), lookupMib=lookup_mib, lexicographicMode=False)

    for (errorIndication, errorStatus, errorIndex, varBinds) in next_cmd:
        if errorIndication or errorStatus or errorIndex:
            return
        for varBind in varBinds:
            res.append([x.prettyPrint() for x in varBind])
    return dict(res)

二、调用common.py模块编写具体实现代码

from pprint import pprint
from common import snmp_walk


def snmp_get_intf_info(ip, community):
    res = {}
    if_name = snmp_walk(ip, community, "1.3.6.1.2.1.31.1.1.1.1")
    if_status = snmp_walk(ip, community, "1.3.6.1.2.1.2.2.1.8")
    if if_name:
        res.update({v: {'index': k.split('.')[-1]} for k, v in if_name.items()})
    if if_status and res:
        tmp_dict = {k.split('.')[-1]: 'up' if v == '1' else 'down' for k, v in if_status.items()}
        for k, v in res.items():
            res[k]['if_status'] = tmp_dict.get(res[k]['index'])
    return res


if __name__ == "__main__":
    pprint(snmp_get_intf_info("10.0.0.3", "Huawei@123"))

三、输出结果为字典格式:

{'GE1/0/0': {'if_status': 'up', 'index': '5'},
 'GE1/0/1': {'if_status': 'down', 'index': '6'},
 'GE1/0/2': {'if_status': 'down', 'index': '7'},
 'GE1/0/3': {'if_status': 'down', 'index': '8'},
 'GE1/0/4': {'if_status': 'down', 'index': '9'},
 'GE1/0/5': {'if_status': 'down', 'index': '10'},
 'GE1/0/6': {'if_status': 'down', 'index': '11'},
 'GE1/0/7': {'if_status': 'down', 'index': '12'},
 'GE1/0/8': {'if_status': 'down', 'index': '13'},
 'GE1/0/9': {'if_status': 'down', 'index': '14'},
 'InLoopBack0': {'if_status': 'up', 'index': '3'},
 'LoopBack0': {'if_status': 'up', 'index': '15'},
 'MEth0/0/0': {'if_status': 'down', 'index': '4'},
 'NULL0': {'if_status': 'up', 'index': '2'}}