Python 封装SNMP调用接口

PySNMP 是一个纯粹用Python实现的SNMP,用PySNMP的最抽象的API为One-line Applications,其中有两类API：同步的和非同步的,都在模块pysnmp.entity.rfc3413.oneliner.cmdgen 中实现,如下是Get方式与Walk方式的基本实现.
首先需要在系统中安装SNMP客户端,对于Linux平台来说只需要执行如下配置过程即可.

1. [root@localhost ~]# yum install -y net-snmp
2. [root@localhost ~]# cat /etc/snmp/snmpd.conf |grep -vE "^#|^$"
3. com2sec notConfigUser  default       public
5. group   notConfigGroup v1           notConfigUser
6. group   notConfigGroup v2c           notConfigUser
8. view    systemview    included   .1
9. view    systemview    included   .1
11. access  notConfigGroup  ""  any  noauth  exact  systemview none none
13. [root@localhost ~]# systemctl restart snmpd
14. [root@localhost ~]# systemctl enable snmpd

复制代码

如果是Windows系统则需要在客户机服务列表，开启SNMP支持，并设置好一个团体名称，如下图。

当我们配置好客户端后，服务端就客户获取数据了，我们以一个OID序号为例，我们查询特定序号对应的名称，然后将其记录下来，例如下面这样。

1. C:\Users\admin> snmpwalk -v 2c -c public 192.168.1.101 .1.3.6.1.2.1.25.2.2
2. HOST-RESOURCES-MIB::hrMemorySize.0 = INTEGER: 2096632 KBytes

复制代码

首先我们不使用PySNMP模块直接开线程调用看看,该代码如下所示.

1. import os,re,time
3. # 通过SNMP收集主机CPU利用率: 通过SNMP协议,收集目标主机的CPU利用率(百分比),并返回JSON字符串.
4. def Get_CPU_Info(addr):
5.     try:
6.         Head = ["HostName","CoreLoad","CpuUser","CpuSystem","CpuIdle"]
7.         CPU = []
8.         ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " .1.3.6.1.2.1.1.5")
9.         CPU.append(ret.read().split(":")[3].strip())
10.         ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " .1.3.6.1.2.1.25.3.3.1.2")
11.         CPU.append(ret.read().split(":")[3].strip())
13.         for i in [9,10,11]:
14.             ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " 1.3.6.1.4.1.2021.11.{}.0".format(i))
15.             ret = ret.read()
16.             Info = ret.split(":")[3].strip()
17.             CPU.append(Info)
18.         return dict(zip(Head,CPU))
19.     except Exception:
20.         return 0
22. # 通过SNMP获取系统CPU负载信息: 分别获取到系统的1,5,15分钟的负载信息,并返回JSON格式.
23. def Get_Load_Info(addr):
24.     try:
25.         Head = ["HostName","Load1","Load5","Load15"]
26.         SysLoad = []
27.         ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " .1.3.6.1.2.1.1.5")
28.         SysLoad.append(ret.read().split(":")[3].strip())
30.         ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " .1.3.6.1.4.1.2021.10.1.3")
31.         load = list(re.sub(".*STRING: ", "", ret.read()).split("\n"))
32.         SysLoad.append(load[0])
33.         SysLoad.append(load[1])
34.         SysLoad.append(load[2])
35.         return dict(zip(Head,SysLoad))
36.     except Exception:
37.         return 0
39. # 通过SNMP获取系统内存占用: 内存利用率，获取到之后，将其转化为字典格式保存。
40. def Get_Mem_Info(addr):
41.     try:
42.         Head = ["HostName","memTotalSwap","memAvailSwap","memTotalReal","memTotalFree"]
43.         SysMem = []
44.         ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " .1.3.6.1.2.1.1.5")
45.         SysMem.append(ret.read().split(":")[3].strip())
46.         ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " .1.3.6.1.4.1.2021.4")
47.         mem = ret.read().split("\n")
48.         for i in [2,3,4,6]:
49.             SysMem.append(re.sub(".*INTEGER: ","",mem[i]).split(" ")[0])
50.         return dict(zip(Head,SysMem))
51.     except Exception:
52.         return 0
54. # 通过SNMP获取系统磁盘数据: 这个案例并不完整,我只写了一点,后面有个问题一直没有解决.
55. def Get_Disk_Info(addr):
56.     try:
57.         dic = {}
58.         list = []
59.         ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " HOST-RESOURCES-MIB::hrStorageDescr")
60.         DiskName = ret.read().split("\n")
61.         ret =os.popen("snmpwalk -v 2c -c nmap " + addr + " HOST-RESOURCES-MIB::hrStorageUsed")
62.         DiskUsed = ret.read().split("\n")
63.         ret = os.popen("snmpwalk -v 2c -c nmap " + addr + " HOST-RESOURCES-MIB::hrStorageSize")
64.         DiskSize = ret.read().split("\n")
66.         for i in range(1,len(DiskName) - 7):
67.             dic["Name"]= DiskName[i + 5].split(":")[3]
68.             dic["Used"]= DiskUsed[i + 5].split(":")[3]
69.             dic["Size"]= DiskSize[i + 5].split(":")[3]
70.             list.append(dic)
71.         return list
72.     except Exception:
73.         return 0
75. if __name__ == '__main__':
76.     for i in range(100):
77.         dic = Get_CPU_Info("192.168.1.20")
78.         print(dic)
79.         time.sleep(1)

复制代码

我们使用pysnmp模块来做，安装pysnmp很简单，执行命令pip install pysnmp 即可，安装后使用以下代码执行即可获取到目标数据，获取方式分为两种一种为Get另一种为Walk.

1. from pysnmp.hlapi import *
2. import os,sys
4. class NetSNMP():
5.     def __init__(self,address,region):
6.         self.region = region
7.         self.address = address
9.     # 获取指定数据的方法
10.     def GetNumber(self,oid,sub_oid,sub_id):
11.         iterator = getCmd(SnmpEngine(),
12.                           CommunityData(self.region),
13.                           UdpTransportTarget((self.address, 161)),
14.                           ContextData(),
15.                           ObjectType(ObjectIdentity(oid, sub_oid, sub_id)))
16.         errorIndication, errorStatus, errorIndex, varBinds = next(iterator)
18.         if errorIndication:
19.             return False
20.         else:
21.             if errorStatus:
22.                 return False
23.             else:
24.                 for varBind in varBinds:
25.                     return [x.prettyPrint() for x in varBind]
27.     # 使用Walk拉取数据
28.     def WalkNumber(self, oid):
29.         res = []
30.         for (errorIndication, errorStatus, errorIndex, varBinds) in nextCmd(SnmpEngine(),
31.              CommunityData(self.region),UdpTransportTarget((self.address, 161)),ContextData(),
32.              ObjectType(ObjectIdentity(oid)).addMibSource(
33.              './site-packages/pysnmp/smi/mibs','pysnmp_mibs'),lexicographicMode=False):
34.             if errorIndication:
35.                 print(errorIndication, file=sys.stderr)
36.                 break
37.             elif errorStatus:
38.                 print('%s at %s' % (errorStatus.prettyPrint(),
39.                                     errorIndex and varBinds[int(errorIndex) - 1][0] or '?'),
40.                       file=sys.stderr)
41.                 break
42.             else:
43.                 for varBind in varBinds:
44.                     res.append(str(varBind))
45.         return res
47. if __name__ == "__main__":
49.     # 初始化
50.     ptr = NetSNMP("192.168.81.130","public")
52.     # 使用GET方式获取OID数据
53.     ret = ptr.GetNumber("HOST-RESOURCES-MIB","hrMemorySize",0)
54.     print("类型: {} --> 返回结果: {} --> 解析: {}".format(type(ret),ret,ret[1]))
56.     # 使用Walk方式获取OID数据
57.     ret = ptr.WalkNumber(".1.3.6.1.2.1.2.2.1.6")
58.     for each in ret:
59.         mac = each.split("=")[1]
60.         if len(mac) > 1:
61.             print("网卡接口: {}".format(mac))

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！