bb6ca556c7
Add a new module "disk" that takes an optional parameter (the path) and displays free & total disk space, along with the usage percentage. Also, added Tunnel/VPN support to the themeing of the "net" module.
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
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 = []
|
|
|
|
try:
|
|
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"
|
|
except Exception as e:
|
|
self._state = "down"
|
|
addr = []
|
|
|
|
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["wlan{}".format(intf)]
|
|
|
|
def _istunnel(self, intf):
|
|
return intf.startswith("tun")
|
|
|
|
def state(self):
|
|
t = "wireless" if self._iswlan(self._intf) else "wired"
|
|
|
|
t = "tunnel" if self._istunnel(self._intf) else t
|
|
|
|
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
|