Display strength of wireless device/networks

2 files changed, 31 insertions, 2 deletions

diff --git a/shell/view/devices/wirelessnetwork.py b/shell/view/devices/wirelessnetwork.py
index 659ec6b..a08de58 100644
--- a/shell/view/devices/wirelessnetwork.py
+++ b/shell/view/devices/wirelessnetwork.py
@@ -3,4 +3,23 @@ from view.devices import deviceview
 class DeviceView(deviceview.DeviceView):
     def __init__(self, model):
         deviceview.DeviceView.__init__(self, model)
-        self.props.icon_name = 'theme:stock-net-wireless-00'
+        self._model = model
+
+        self._update_icon()
+        model.connect('notify::strength', self._strength_changed_cb)
+
+    def _strength_changed_cb(self, model, pspec):
+        self._update_icon()
+
+    def _update_icon(self):
+        strength = self._model.props.strength
+        if strength < 21:
+            self.props.icon_name = 'theme:stock-net-wireless-00'
+        elif strength < 41:
+            self.props.icon_name = 'theme:stock-net-wireless-21-40'
+        elif strength < 61:
+            self.props.icon_name = 'theme:stock-net-wireless-41-60'
+        elif strength < 81:
+            self.props.icon_name = 'theme:stock-net-wireless-61-80'
+        else:
+            self.props.icon_name = 'theme:stock-net-wireless-81-100'
diff --git a/shell/view/home/MeshBox.py b/shell/view/home/MeshBox.py
index 55d8cce..707ad6e 100644
--- a/shell/view/home/MeshBox.py
+++ b/shell/view/home/MeshBox.py
@@ -42,7 +42,17 @@ class AccessPointView(CanvasIcon):
         network_manager.set_active_device(device, network)
 
     def _update_icon(self):
-        self.props.icon_name = 'theme:stock-net-wireless-00'
+        strength = self._model.props.strength
+        if strength < 21:
+            self.props.icon_name = 'theme:stock-net-wireless-00'
+        elif strength < 41:
+            self.props.icon_name = 'theme:stock-net-wireless-21-40'
+        elif strength < 61:
+            self.props.icon_name = 'theme:stock-net-wireless-41-60'
+        elif strength < 81:
+            self.props.icon_name = 'theme:stock-net-wireless-61-80'
+        else:
+            self.props.icon_name = 'theme:stock-net-wireless-81-100'
 
 class ActivityView(SnowflakeBox):
     def __init__(self, shell, menu_shell, model):