Fix memory leak in netatmo (#49464)

2021-04-19 23:03:07 -10:00
parent 12853438c5
commit b557d20fbb
1 changed files with 3 additions and 1 deletions
--- a/homeassistant/components/netatmo/init.py
+++ b/homeassistant/components/netatmo/init.py
@@ -188,7 +188,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
        except pyatmo.ApiError as err:
            _LOGGER.error("Error during webhook registration - %s", err)

-        hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, unregister_webhook)
+        entry.async_on_unload(
+            hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, unregister_webhook)
+        )

    if hass.state == CoreState.running:
        await register_webhook(None)