fix: assign hubitat hass data before use

Some of HA's recent performance improvements exposed a race condition in the integration's initialization code.
jason0x43 · Apr 27, 2024 · 15012c7 · 15012c7
1 parent 08c2ac5
commit 15012c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/hubitat/__init__.py b/custom_components/hubitat/__init__.py
@@ -34,11 +34,11 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
 
     hub = Hub(hass, config_entry, len(hass.data[DOMAIN]) + 1)
 
+    hass.data[DOMAIN][config_entry.entry_id] = hub
+
     if not await hub.async_setup():
         return False
 
-    hass.data[DOMAIN][config_entry.entry_id] = hub
-
     hub.async_update_device_registry()
 
     async_register_services(hass, config_entry)