Fix long Hass reboot

AlexxIT · Feb 25, 2021 · d963821 · d963821
1 parent e8423c0
commit d963821
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/custom_components/dataplicity/__init__.py b/custom_components/dataplicity/__init__.py
@@ -1,6 +1,7 @@
 from threading import Thread
 
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant
 from homeassistant.requirements import async_process_requirements
 from homeassistant.util import package
@@ -39,6 +40,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
     client.port_forward.add_service('web', hass.config.api.port)
     Thread(name=DOMAIN, target=client.run_forever).start()
 
+    async def hass_stop(event):
+        client.exit()
+
+    hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, hass_stop)
+
     return True