Fix #6 with updated icons.

djtimca · Jun 3, 2022 · 192561d · 192561d
1 parent a1c242a
commit 192561d
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/custom_components/omnilogic/switch.py b/custom_components/omnilogic/switch.py
@@ -74,10 +74,10 @@ def __init__(
         state_key: str,
     ) -> None:
 
-        switch_type = coordinator.data[item_id].get("Type")
+        switch_type = coordinator.data[item_id].get("Type", "")
 
-        if switch_type:
-            icon = SWITCH_ICONS[switch_type]
+        if switch_type == "RLY_VALVE_ACTUATOR":
+            icon = "mdi:valve"
 
         """Initialize Entities."""
         super().__init__(
@@ -273,7 +273,3 @@ async def async_set_speed(self, speed):
         }
     ],
 }
-
-SWITCH_ICONS = {
-    "RLY_VALVE_ACTUATOR": "mdi:valve"
-}