From 1efe8db021a89fb6e1e377466715ffdd75268118 Mon Sep 17 00:00:00 2001 From: rospogrigio Date: Wed, 8 Jun 2022 14:08:48 +0200 Subject: [PATCH] Fixed restoring entity state when reloading --- custom_components/localtuya/__init__.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/custom_components/localtuya/__init__.py b/custom_components/localtuya/__init__.py index 3b4ce31..f1af7f0 100644 --- a/custom_components/localtuya/__init__.py +++ b/custom_components/localtuya/__init__.py @@ -288,11 +288,6 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry): """Unload a config entry.""" platforms = {} - hass.data[DOMAIN][entry.entry_id][UNSUB_LISTENER]() - for dev_id, device in hass.data[DOMAIN][TUYA_DEVICES].items(): - if device.connected: - await device.close() - for dev_id, dev_entry in entry.data[CONF_DEVICES].items(): for entity in dev_entry[CONF_ENTITIES]: platforms[entity[CONF_PLATFORM]] = True @@ -305,6 +300,12 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry): ] ) ) + + hass.data[DOMAIN][entry.entry_id][UNSUB_LISTENER]() + for dev_id, device in hass.data[DOMAIN][TUYA_DEVICES].items(): + if device.connected: + await device.close() + if unload_ok: hass.data[DOMAIN][TUYA_DEVICES] = {}