4e648cf009
Big oversight in my previous commits: Widgets need to be able to have specific configurations (i.e. the path for different instances of the "disk" module has to be different). To account for that, it is now possible to assign an "alias" to a module instance using ":" (for example: -m "disk:home"). This alias is then used for the configuration parameter resolution automatically, for example: -m disk:home -p home.path=/home As a consequence, parameter names in the module code are now relative to the module, which means: shorter!
73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
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
|