misc

Signed-off-by: Mattias Andrée <maandree@operamail.com>

1 files changed, 71 insertions, 0 deletions

diff --git a/src/plugins/snmp.py b/src/plugins/snmp.py
new file mode 100644
index 0000000..dd73583
--- /dev/null
+++ b/src/plugins/snmp.py
@@ -0,0 +1,71 @@
+# -*- python -*-
+'''
+xpybar – xmobar replacement written in python
+Copyright © 2014  Mattias Andrée (maandree@member.fsf.org)
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+'''
+
+from util import *
+
+
+class SNMP:
+    '''
+    IPv4 SNMP data
+    '''
+    
+    
+    def __init__(self):
+        '''
+        Constructor
+        '''
+        snmp = None
+        with open('/proc/net/snmp', 'rb') as file:
+            snmp = file.read()
+        snmp = snmp.decode('utf-8', 'replace')
+        
+        filter_ = lambda array : filter(lambda x : not x == '', array)
+        snmp = list(map(lambda x : x.split(' '), filter_(snmp.split('\n'))))
+        snmp_h = filter(lambda i : i % 2 == 0, range(len(snmp)))
+        snmp_d = filter(lambda i : i % 2 == 0, range(len(snmp)))
+        snmp_h = list(map(lambda i : snmp[i], snmp_h))
+        snmp_d = list(map(lambda i : snmp[i], snmp_d))
+        snmp = zip(snmp_h, snmp_d)
+        
+        self.__info = {}
+        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)
+    
+    
+    def __contains__(self, key):
+        '''
+        Get whether or not a key is available
+        
+        @param   key:str  The key
+        @return  :bool    The availability
+        '''
+        return key in self.__info
+    
+    
+    def __getitem__(self, key):
+        '''
+        Lookup a field from '/proc/net/snmp'
+        
+        @param   key:str  The field name
+        @return  :int     The value of the field
+        '''
+        return self.__info[key]
+