- Fix "RuntimeError: Detected that custom integration ..."

- Immediately update sensors when starting, instead of waiting 5 minutes.
Breina · May 2, 2024 · 466e442 · 466e442
1 parent e068436
commit 466e442
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/custom_components/idrac_power/binary_sensor.py b/custom_components/idrac_power/binary_sensor.py
@@ -76,4 +76,4 @@ def name(self):
 
     def update_value(self, status: bool):
         self._attr_is_on = status
-        self.async_schedule_update_ha_state()
+        self.schedule_update_ha_state()
diff --git a/custom_components/idrac_power/sensor.py b/custom_components/idrac_power/sensor.py
@@ -66,11 +66,16 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_e
     async def refresh_sensors_task():
         while True:
             _LOGGER.debug("Refreshing sensors")
-            await hass.async_add_executor_job(rest_client.update_thermals)
-            await hass.async_add_executor_job(rest_client.update_status)
-            await hass.async_add_executor_job(rest_client.update_power_usage)
+            await update_all()
             await asyncio.sleep(rest_client.interval)
 
+    async def update_all():
+        await hass.async_add_executor_job(rest_client.update_thermals)
+        await hass.async_add_executor_job(rest_client.update_status)
+        await hass.async_add_executor_job(rest_client.update_power_usage)
+
+    await update_all()
+
     hass.async_create_background_task(refresh_sensors_task(), f"Update {name} iDRAC task")
 
 
@@ -100,7 +105,7 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name):
 
     def update_value(self, new_value: int):
         self._attr_native_value = new_value
-        self.async_schedule_update_ha_state()
+        self.schedule_update_ha_state()
 
 
 class IdracFanSensor(SensorEntity):
@@ -127,7 +132,7 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name, index):
 
     def update_value(self, thermal: dict):
         self._attr_native_value = thermal['Fans'][self.index]['Reading']
-        self.async_schedule_update_ha_state()
+        self.schedule_update_ha_state()
 
 
 class IdracTempSensor(SensorEntity):
@@ -154,4 +159,4 @@ def __init__(self, hass, rest: IdracRest, device_info, unique_id, name, index):
 
     def update_value(self, thermal: dict):
         self._attr_native_value = thermal['Temperatures'][self.index]['ReadingCelsius']
-        self.async_schedule_update_ha_state()
+        self.schedule_update_ha_state()