90 lines
3.1 KiB
Python
90 lines
3.1 KiB
Python
"""
|
|
A module to show currently active network connection (ethernet or wifi) and connection strength if the connection is wireless.
|
|
|
|
Dependencies: nm-connection-editor if users would like a graphical
|
|
network manager when left-clicking the module
|
|
"""
|
|
|
|
|
|
import util.cli
|
|
import util.format
|
|
|
|
import core.module
|
|
import core.widget
|
|
import core.input
|
|
|
|
|
|
class Module(core.module.Module):
|
|
@core.decorators.every(seconds=10)
|
|
def __init__(self, config, theme):
|
|
super().__init__(config, theme, core.widget.Widget(self.network))
|
|
self.__is_wireless = False
|
|
self.__interface = None
|
|
self.__message = None
|
|
self.__signal = -110
|
|
|
|
# Set up event handler for left mouse click
|
|
core.input.register(
|
|
self, button=core.input.LEFT_MOUSE, cmd="nm-connection-editor"
|
|
)
|
|
|
|
# Get network information to display to the user
|
|
def network(self, widgets):
|
|
# run ip route command, tokenize output
|
|
cmd = "ip route get 8.8.8.8"
|
|
std_out = util.cli.execute(cmd)
|
|
route_str = " ".join(std_out.split())
|
|
route_tokens = route_str.split(" ")
|
|
|
|
# Attempt to extract a valid network interface device
|
|
try:
|
|
self.__interface = route_tokens[route_tokens.index("dev") + 1]
|
|
except ValueError:
|
|
self.__interface = None
|
|
|
|
# Check to see if the interface (if it exists) is wireless
|
|
if self.__interface:
|
|
with open("/proc/net/wireless", "r") as f:
|
|
self.__is_wireless = self.__interface in f.read()
|
|
f.close()
|
|
|
|
# setup message to send to the user
|
|
if self.__interface is None:
|
|
self.__message = " No connection"
|
|
elif self.__is_wireless:
|
|
iw_dat = util.cli.execute("iwgetid")
|
|
has_ssid = "ESSID" in iw_dat
|
|
ssid = (
|
|
iw_dat[iw_dat.index(":") + 1 :].replace('"', "").strip()
|
|
if has_ssid
|
|
else "Unknown"
|
|
)
|
|
|
|
# Get connection strength
|
|
cmd = "iwconfig {}".format(self.__interface)
|
|
config_dat = " ".join(util.cli.execute(cmd).split())
|
|
config_tokens = config_dat.replace("=", " ").split()
|
|
strength = config_tokens[config_tokens.index("level") + 1]
|
|
self.__signal = util.format.asint(strength, minimum=-110, maximum=-30)
|
|
|
|
self.__message = self.__generate_wireles_message(ssid, self.__signal)
|
|
else:
|
|
# Set signal to -30 as ethernet shouldn't have signal issues
|
|
self.__signal = -30
|
|
self.__message = " Ethernet"
|
|
|
|
return self.__message
|
|
|
|
# The signal is measured in decibels/milliwatt, hence the weird numbers
|
|
def state(self, widget):
|
|
if self.__signal < -80:
|
|
return "critical"
|
|
if self.__signal < -65:
|
|
return "warning"
|
|
|
|
return None
|
|
|
|
# manually done for better granularity / ease of parsing strength data
|
|
def __generate_wireles_message(self, ssid, strength):
|
|
computed_strength = 100 * ((strength + 100) / 70.0)
|
|
return " {} {}%".format(ssid, int(computed_strength))
|