[modules] Add NIC module

Add a module that displays the status of all NICs (interface name, list
of IPs and state).

In its status, it also exposes whether it's a WiFi or a wired NIC.

For this functionality, additional code was implemented to allow a
module to add multiple elements to the bar at once. The framework calls
the module until its "next()" method return False.

bumblebee/module.py

@@ -15,4 +15,7 @@ class Module(object):
     def state(self):
         return "default"
 
+    def next(self):
+        return False
+
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4

bumblebee/modules/nic.py

@@ -0,0 +1,61 @@
+import pyroute2
+import netifaces
+import bumblebee.module
+
+class Module(bumblebee.module.Module):
+    def __init__(self, args):
+        super(Module, self).__init__(args)
+        self._exclude = ( "lo", "virbr" )
+        self._interfaces = [ i for i in netifaces.interfaces() if not i.startswith(self._exclude) ]
+        self._index = 0
+        self._intf = self._interfaces[0] if len(self._interfaces) > 0 else None
+        self._cache = {}
+        self._state = "down"
+
+    def data(self):
+        if len(self._interfaces) <= self._index:
+            return "n/a"
+        self._intf = self._interfaces[self._index]
+        self._state = "down"
+        addr = []
+
+        if netifaces.AF_INET in netifaces.ifaddresses(self._intf):
+            for ip in netifaces.ifaddresses(self._intf)[netifaces.AF_INET]:
+                if "addr" in ip and ip["addr"] != "":
+                    addr.append(ip["addr"])
+                    self._state = "up"
+
+        return "{} {} {}".format(self._intf, self._state, ", ".join(addr))
+
+    def next(self):
+        self._index += 1
+        if self._index < len(self._interfaces):
+            return True
+        self._index = 0
+        # reload to support hotplug
+        self._interfaces = [ i for i in netifaces.interfaces() if not i.startswith(self._exclude) ]
+        return False
+
+    def _iswlan(self, intf):
+        if not "wlan{}".format(intf) in self._cache:
+            iw = pyroute2.IW()
+            ip = pyroute2.IPRoute()
+            idx = ip.link_lookup(ifname=intf)[0]
+            try:
+                iw.get_interface_by_ifindex(idx)
+                self._cache["wlan{}".format(intf)] = True
+            except exception as e:
+                self._cache["wlan{}".format(intf)] = False
+        return self._cache
+
+    def state(self):
+        t = "wireless" if self._iswlan(self._intf) else "wired"
+        return "{}-{}".format(t, self._state)
+
+    def warning(self):
+        return self._state != "up"
+
+    def critical(self):
+        return self._state == "down"
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4

bumblebee/outputs/i3.py

@@ -15,28 +15,32 @@ class i3bar(bumblebee.output.Output):
     def add(self, obj):
         theme = self.theme()
 
-        data = {
+        while True:
-            u"full_text": "{}{}{}".format(theme.prefix(obj), obj.data(), theme.suffix(obj)),
+            data = {
-            "color": theme.color(obj),
+                u"full_text": "{}{}{}".format(theme.prefix(obj), obj.data(), theme.suffix(obj)),
-            "background": theme.background(obj),
+                "color": theme.color(obj),
-        }
+                "background": theme.background(obj),
+            }
 
-        if theme.urgent(obj) and obj.critical():
+            if theme.urgent(obj) and obj.critical():
-            data["urgent"] = True
+                data["urgent"] = True
 
-        if theme.default_separators(obj) == False:
+            if theme.default_separators(obj) == False:
-            data["separator"] = False
+                data["separator"] = False
-            data["separator_block_width"] = 0
+                data["separator_block_width"] = 0
-            if theme.separator(obj):
+                if theme.separator(obj):
-                self._data.append({
+                    self._data.append({
-                    u"full_text": theme.separator(obj),
+                        u"full_text": theme.separator(obj),
-                    "color": theme.background(obj),
+                        "color": theme.background(obj),
-                    "background": theme.previous_background(),
+                        "background": theme.previous_background(),
-                    "separator": False,
+                        "separator": False,
-                    "separator_block_width": 0,
+                        "separator_block_width": 0,
-                })
+                    })
 
-        self._data.append(data)
+            self._data.append(data)
+            if obj.next() == False:
+                break
+            theme.next()
 
     def get(self):
         data = json.dumps(self._data)

bumblebee/themes/solarized-powerline.json

@@ -32,6 +32,22 @@
 	"cpu": {
 		"prefix": "  "
 	},
+	"nic": {
+		"states": {
+			"wireless-up": {
+				"prefix": "  "
+			},
+			"wireless-down": {
+				"prefix": "  "
+			},
+			"wired-up": {
+				"prefix": "  "
+			},
+			"wired-down": {
+				"prefix": "  "
+			}
+		}
+	},
 	"battery": {
 		"states": {
 			"charged": {