85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import subprocess
|
|
|
|
import psutil
|
|
from fabric import Fabricator
|
|
from fabric.widgets.box import Box
|
|
from snippets import MaterialIcon
|
|
|
|
|
|
class NetworkIndicator(Box):
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.update_interval = 5000 # Increased interval to 5 seconds
|
|
self.last_status = None
|
|
self.wifi_strength_cache = None
|
|
self.fabricator = Fabricator(
|
|
interval=self.update_interval,
|
|
poll_from=self.update_network_status,
|
|
)
|
|
|
|
def update_network_status(self, *_):
|
|
current_status = self.get_network_status()
|
|
if current_status != self.last_status:
|
|
self.update_icon(current_status)
|
|
self.last_status = current_status
|
|
return True
|
|
|
|
def update_icon(self, current_status):
|
|
self.icon_label = MaterialIcon(current_status, size=16)
|
|
self.children = (self.icon_label,)
|
|
|
|
def get_network_status(self):
|
|
if self.is_ethernet_connected():
|
|
return "lan"
|
|
if self.is_wifi_connected():
|
|
return self.get_wifi_strength_icon()
|
|
return "signal_wifi_off"
|
|
|
|
def get_wifi_strength_icon(self):
|
|
wifi_strength = self.check_wifi_strength()
|
|
if wifi_strength is not None:
|
|
self.wifi_strength_cache = wifi_strength # Cache the result
|
|
else:
|
|
wifi_strength = self.wifi_strength_cache # Use cached value if available
|
|
|
|
return self.wifi_signal_icon(wifi_strength)
|
|
|
|
def wifi_signal_icon(self, strength):
|
|
if strength is None or strength <= -100:
|
|
return "signal_wifi_0_bar"
|
|
if strength > -50:
|
|
return "signal_wifi_4_bar"
|
|
if strength > -60:
|
|
return "network_wifi_3_bar"
|
|
if strength > -70:
|
|
return "network_wifi_2_bar"
|
|
return "network_wifi_1_bar"
|
|
|
|
def check_wifi_strength(self):
|
|
try:
|
|
result = subprocess.run(["iwconfig"], capture_output=True, text=True)
|
|
for line in result.stdout.splitlines():
|
|
if "Signal level" in line:
|
|
signal_level = line.split("Signal level=")[1].split(" ")[0]
|
|
return int(signal_level.strip().replace("dBm", ""))
|
|
return None
|
|
except Exception:
|
|
return None
|
|
|
|
def is_wifi_connected(self):
|
|
return any(
|
|
self._is_interface_up(interface, ["wlan", "wifi"])
|
|
for interface in psutil.net_if_addrs()
|
|
)
|
|
|
|
def is_ethernet_connected(self):
|
|
return any(
|
|
self._is_interface_up(interface, ["enp", "eth"])
|
|
for interface in psutil.net_if_addrs()
|
|
)
|
|
|
|
def _is_interface_up(self, interface, keywords):
|
|
if any(substring in interface.lower() for substring in keywords):
|
|
return psutil.net_if_stats().get(interface, {}).isup
|
|
return False
|