db8737a6aa
Repeatedly querying the interface type the way I'm doing right now results in "too many files open" pretty quickly. Obviously, it's a bit of a concern that I am leaking a file descriptor somewhere, but for now, the quickfix is to cache the type (it shouldn't change, anyhow).
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):
|
|
super(Module, self).__init__(output, config)
|
|
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
|