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