Changing icon for switches to power-plug

dvd-dev · Jan 16, 2022 · 41c9ffa · 41c9ffa
1 parent a72a3cd
commit 41c9ffa
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/custom_components/hilo/switch.py b/custom_components/hilo/switch.py
@@ -31,6 +31,14 @@ def __init__(self, hilo: Hilo, device):
     def state(self):
         return self._device.state
 
+    @property
+    def icon(self):
+        if not self._device.available:
+            return "mdi:lan-disconnect"
+        if self.state == "on":
+            return "mdi:power-plug"
+        return "mdi:power-plug-off"
+
     @property
     def is_on(self):
         return self._device.get_value("is_on")