Fix excepiton on unload domain

Fix for #435
AlexxIT · Feb 3, 2024 · 4d6824d · 4d6824d
1 parent a8b67d8
commit 4d6824d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/custom_components/yandex_station/__init__.py b/custom_components/yandex_station/__init__.py
@@ -171,9 +171,12 @@ async def update_cookie_and_token(**kwargs):
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
     quasar: YandexQuasar = hass.data[DOMAIN][entry.unique_id]
     quasar.stop()
-    return await hass.config_entries.async_unload_platforms(
-        entry, MAIN_DOMAINS + SUB_DOMAINS
-    )
+
+    domains = MAIN_DOMAINS
+    if CONF_INCLUDE in hass.data[DOMAIN][DATA_CONFIG]:
+        domains += SUB_DOMAINS  
+
+    return await hass.config_entries.async_unload_platforms(entry, domains)
 
 
 async def _init_local_discovery(hass: HomeAssistant):