cloudapi: support updating device_config data

Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
Noltari · Jul 25, 2024 · d8b5cea · d8b5cea
1 parent 3ebb65b
commit d8b5cea
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/aioairzone_cloud/cloudapi.py b/aioairzone_cloud/cloudapi.py
@@ -666,8 +666,13 @@ def set_system_zones_data(self, system: System) -> None:
     async def update_aidoo(self, aidoo: Aidoo) -> None:
         """Update Airzone Cloud Zone from API."""
         device_data = await self.api_get_device_status(aidoo)
+        if self.options.device_config:
+            device_config = await self.api_get_device_config(aidoo)
+            print(device_config)
+        else:
+            device_config: dict[str, Any] = {}
 
-        update = EntityUpdate(UpdateType.API_FULL, device_data)
+        update = EntityUpdate(UpdateType.API_FULL, device_config | device_data)
 
         await aidoo.update(update)
 
@@ -773,8 +778,13 @@ def get_ws_device_data(self, device: Device) -> dict[str, Any] | None:
     async def update_system(self, system: System) -> None:
         """Update Airzone Cloud System from API."""
         device_data = await self.api_get_device_status(system)
+        if self.options.device_config:
+            device_config = await self.api_get_device_config(system)
+            print(device_config)
+        else:
+            device_config: dict[str, Any] = {}
 
-        update = EntityUpdate(UpdateType.API_FULL, device_data)
+        update = EntityUpdate(UpdateType.API_FULL, device_config | device_data)
 
         await system.update(update)
 
@@ -858,8 +868,13 @@ async def update_webservers(self, devices: bool) -> None:
     async def update_zone(self, zone: Zone) -> None:
         """Update Airzone Cloud Zone from API."""
         device_data = await self.api_get_device_status(zone)
+        if self.options.device_config:
+            device_config = await self.api_get_device_config(zone)
+            print(device_config)
+        else:
+            device_config: dict[str, Any] = {}
 
-        update = EntityUpdate(UpdateType.API_FULL, device_data)
+        update = EntityUpdate(UpdateType.API_FULL, device_config | device_data)
 
         await zone.update(update)
 

diff --git a/aioairzone_cloud/common.py b/aioairzone_cloud/common.py
@@ -13,6 +13,7 @@ class ConnectionOptions:
 
     username: str
     password: str
+    device_config: bool = False
     websockets: bool = True