阅读量:2
在网络自动化过程中, 通过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'}} 