Handle entity unload (#337)

* hande entity unload. fixes #332 Co-authored-by: Joshua Mulliken <[email protected]>
SecKatie · Feb 4, 2022 · 6772130 · 6772130
1 parent bfa3c68
commit 6772130
Show file tree

Hide file tree

Showing 7 changed files with 186 additions and 211 deletions.
diff --git a/custom_components/wyzeapi/climate.py b/custom_components/wyzeapi/climate.py
@@ -321,4 +321,4 @@ async def async_added_to_hass(self) -> None:
         return await super().async_added_to_hass()
 
     async def async_will_remove_from_hass(self) -> None:
-        self._thermostat_service.unregister_updater()
+        self._thermostat_service.unregister_updater(self._thermostat)
diff --git a/custom_components/wyzeapi/light.py b/custom_components/wyzeapi/light.py
@@ -255,4 +255,4 @@ async def async_added_to_hass(self) -> None:
         return await super().async_added_to_hass()
 
     async def async_will_remove_from_hass(self) -> None:
-        self._bulb_service.unregister_updater()
+        self._bulb_service.unregister_updater(self._bulb)
diff --git a/custom_components/wyzeapi/lock.py b/custom_components/wyzeapi/lock.py
@@ -172,4 +172,4 @@ async def async_added_to_hass(self) -> None:
         return await super().async_added_to_hass()
 
     async def async_will_remove_from_hass(self) -> None:
-        self._lock_service.unregister_updater()
+        self._lock_service.unregister_updater(self._lock)
diff --git a/custom_components/wyzeapi/manifest.json b/custom_components/wyzeapi/manifest.json
@@ -1,13 +1,13 @@
 {
   "domain": "wyzeapi",
-  "version": "0.1.5-alpha.0",
+  "version": "0.1.5-alpha.1",
   "name": "Wyze",
   "config_flow": true,
   "iot_class": "cloud_polling",
   "documentation": "https://github.com/JoshuaMulliken/ha-wyzeapi#readme",
   "issue_tracker": "https://github.com/JoshuaMulliken/ha-wyzeapi/issues",
   "requirements": [
-    "wyzeapy==0.5.6a0"
+    "wyzeapy==0.5.6a1"
   ],
   "codeowners": [
     "@JoshuaMulliken"

diff --git a/custom_components/wyzeapi/switch.py b/custom_components/wyzeapi/switch.py
@@ -281,4 +281,4 @@ async def async_added_to_hass(self) -> None:
         return await super().async_added_to_hass()
 
     async def async_will_remove_from_hass(self) -> None:
-        self._service.unregister_updater()
+        self._service.unregister_updater(self._device)
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,15 +4,15 @@
 
 [tool.poetry]
 name = "ha-wyzeapi"
-version = "0.1.5-alpha.0"
+version = "0.1.5-alpha.1"
 description = "A Home Assistant integration for Wyze devices"
 authors = ["Joshua Mulliken <[email protected]>"]
 license = "Apache-2.0"
 
 [tool.poetry.dependencies]
 python = "^3.8"
 homeassistant = "^2021.11.5"
-wyzeapy = "0.5.6a0"
+wyzeapy = "0.5.6a1"
 
 [tool.poetry.dev-dependencies]