From 4d6824d10550ef69b5d7e668e23dfdc36f6e696d Mon Sep 17 00:00:00 2001 From: shammysha <65885873+shammysha@users.noreply.github.com> Date: Sat, 3 Feb 2024 06:01:15 +0300 Subject: [PATCH] Fix excepiton on unload domain Fix for https://github.com/AlexxIT/YandexStation/issues/435 --- custom_components/yandex_station/__init__.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/custom_components/yandex_station/__init__.py b/custom_components/yandex_station/__init__.py index 3b93e59..b46ac47 100644 --- 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):