bumblebee-status/bumblebee/modules/nic.py

import pyroute2
import netifaces
import bumblebee.module

def usage():
    return "nic"

def notes():
    return "Interfaces starting with 'lo' or 'virbr' are ignored. Critical if the status of an interface is 'down', Warning if it is anything else but 'up'. Interface status is derived from whether an IP address is available or not."

def description():
    return "Displays the names, IP addresses and status of each available interface."

class Module(bumblebee.module.Module):
    def __init__(self, output, config, alias):
        super(Module, self).__init__(output, config, alias)
        self._exclude = ( "lo", "virbr", "docker", "vboxnet" )
        self._state = "down"
        self._typecache = {}

    def widgets(self):
        result = []
        interfaces = [ i for i in netifaces.interfaces() if not i.startswith(self._exclude) ]
        addr = []
        state = "down"
        for intf in interfaces:
            try:
                if netifaces.AF_INET in netifaces.ifaddresses(intf):
                    for ip in netifaces.ifaddresses(intf)[netifaces.AF_INET]:
                        if "addr" in ip and ip["addr"] != "":
                            addr.append(ip["addr"])
                            state = "up"
            except Exception as e:
                addr = []
            widget = bumblebee.output.Widget(self, "{} {} {}".format(
                intf, state, ", ".join(addr)
            ))
            widget.set("intf", intf)
            widget.set("state", state)
            result.append(widget)

        return result

    def _iswlan(self, intf):
        iw = pyroute2.IW()
        ip = pyroute2.IPRoute()
        idx = ip.link_lookup(ifname=intf)[0]
        try:
            iw.get_interface_by_ifindex(idx)
            return True
        except Exception as e:
            return False

    def _istunnel(self, intf):
        return intf.startswith("tun")

    def state(self, widget):
        intf = widget.get("intf")

        if not intf in self._typecache:
            t = "wireless" if self._iswlan(intf) else "wired"
            t = "tunnel" if self._istunnel(intf) else t
            self._typecache[intf] = t

        return "{}-{}".format(self._typecache[intf], widget.get("state"))

    def warning(self, widget):
        return widget.get("state") != "up"

    def critical(self, widget):
        return widget.get("state") == "down"

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4