localapi: handle more invalid API responses

Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
Noltari · Feb 21, 2024 · a6742e2 · a6742e2
1 parent 4e28aad
commit a6742e2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/aioairzone/localapi.py b/aioairzone/localapi.py
@@ -193,8 +193,10 @@ def update_dhw(self, data: dict[str, Any]) -> None:
         else:
             self.hotwater = HotWater(dhw)
 
-    def update_systems(self, data: dict[str, Any]) -> None:
+    def update_systems(self, data: dict[str, Any] | None) -> None:
         """Gather Systems data."""
+        if data is None:
+            raise APIError(f"update_systems: empty Systems API response")
         api_systems = data.get(API_SYSTEMS)
         if api_systems is None:
             raise APIError(f"update_systems: {API_SYSTEMS} not in API response")
@@ -299,6 +301,8 @@ async def update(self) -> None:
             zone.set_available(False)
 
         hvac = await self.get_hvac()
+        if hvac is None:
+            raise APIError(f"update: empty HVAC API response")
         if self.options.system_id == DEFAULT_SYSTEM_ID:
             hvac_systems = hvac.get(API_SYSTEMS)
             if hvac_systems is None: