snmp, snmp6: add keys

Signed-off-by: Mattias Andrée <maandree@kth.se>

2 files changed, 10 insertions, 1 deletions

diff --git a/src/plugins/snmp.py b/src/plugins/snmp.py
index a52cb4e..90e7554 100644
--- a/src/plugins/snmp.py
+++ b/src/plugins/snmp.py
@@ -23,6 +23,8 @@ from util import *
 class SNMP:
     '''
     IPv4 SNMP data
+    
+    @variable  keys:list<str>  List of avaiable keys
     '''
     
     
@@ -44,10 +46,13 @@ class SNMP:
         snmp = zip(snmp_h, snmp_d)
         
         self.__info = {}
+        self.keys = []
         for header_list, data_list in snmp:
             prefix = header_list[0][:-1]
             for suffix, value in zip(header_list[1:], data_list[1:]):
-                self.__info[prefix + suffix] = int(value)
+                key = prefix + suffix
+                self.__info[key] = int(value)
+                self.keys.append(key)
     
     
     def __contains__(self, key):
diff --git a/src/plugins/snmp6.py b/src/plugins/snmp6.py
index afe6de6..bd2ce1c 100644
--- a/src/plugins/snmp6.py
+++ b/src/plugins/snmp6.py
@@ -23,6 +23,8 @@ from util import *
 class SNMP6:
     '''
     IPv6 SNMP data
+    
+    @variable  keys:list<str>  List of avaiable keys
     '''
     
     
@@ -38,8 +40,10 @@ class SNMP6:
         snmp6 = map(lambda x : filter_(x.split(' ')), filter_(snmp6.split('\n')))
         
         self.__info = {}
+        self.keys = []
         for field, value in snmp6:
             self.__info[field] = int(value)
+            self.keys.append(field)
     
     
     def __contains__(self, key):