Check if 'coordinator' exists in hass.data

Changed icon for night cooling switch and status
toringer · Jun 15, 2023 · b6416db · b6416db
1 parent 3ca3591
commit b6416db
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/custom_components/heru/const.py b/custom_components/heru/const.py
@@ -22,6 +22,7 @@
 ICON_CALENDAR = "mdi:calendar"
 ICON_THERMOSTAT = "mdi:home-thermometer"
 ICON_START = "mdi:ray-start-arrow"
+ICON_COOLING = "mdi:snowflake"
 
 # Platforms
 SENSOR = Platform.SENSOR
@@ -159,7 +160,7 @@
         "modbus_address": "1x00038",
         "address": 37,
         "scale": None,
-        "icon": ICON_ALARM,
+        "icon": ICON_COOLING,
         "unit_of_measurement": None,
         "device_class": None,
         "entity_category": EntityCategory.DIAGNOSTIC,
@@ -401,7 +402,7 @@
         "name": "Night cooling enabled",
         "modbus_address": "4x00019",
         "address": 18,
-        "icon": ICON_SWITCH,
+        "icon": ICON_COOLING,
         "register_type": HOLDING_REGISTERS,
     },
 ]

diff --git a/custom_components/heru/helpers/config_flow.py b/custom_components/heru/helpers/config_flow.py
@@ -28,7 +28,7 @@ async def validate_step_user(
         """Validate step_user"""
 
         coordinator = None
-        if DOMAIN in hass.data:
+        if DOMAIN in hass.data and "coordinator" in hass.data[DOMAIN]:
             coordinator = hass.data[DOMAIN]["coordinator"]
             if coordinator is not None:
                 coordinator.pause()