Updates after testing with home assistant

wefabricate · Jul 12, 2024 · 25c535b · 25c535b
1 parent 076d7a4
commit 25c535b
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 7 deletions.
diff --git a/weheat_backend_client/abstractions/auth.py b/weheat_backend_client/abstractions/auth.py
@@ -1,5 +1,5 @@
-from abc import ABC
-from aiohttp import ClientResponse
+from abc import ABC, abstractmethod
+from aiohttp import ClientSession, ClientResponse
 
 
 class AbstractAuth(ABC):

diff --git a/weheat_backend_client/abstractions/discovery.py b/weheat_backend_client/abstractions/discovery.py
@@ -7,14 +7,16 @@
 
 class HeatPumpDiscovery:
     @dataclass
-    class HeatPump:
+    class HeatPumpInfo:
         uuid: str
         name: str
-        household_name: str
+        model: str
+        sn : str
+        has_dhw: bool = False
 
     @staticmethod
-    def discover(api_url: str, access_token: str) -> list[HeatPump]:
-        discovered_pumps = {}
+    def discover(api_url: str, access_token: str) -> list[HeatPumpInfo]:
+        discovered_pumps = []
 
         config = Configuration(host=api_url, access_token=access_token)
 
@@ -23,6 +25,23 @@ def discover(api_url: str, access_token: str) -> list[HeatPump]:
             response = HeatPumpApi(client).api_v1_heat_pumps_get_with_http_info()
             if response.status_code == 200:
                 for pump in response.data:
-                    discovered_pumps[pump.id] = pump.serial_number
+                    model_string = "BlackBird P80 heat pump"
+                    if pump.model == 1:
+                        model_string = "BlackBird P60 heat pump"
+                    elif pump.model == 2:
+                        model_string = "Sparrow P60 heat pump"
 
+                    dhw = False
+                    if pump.dhw_type is not None and pump.dhw_type == 1:
+                        dhw = True
+
+                    discovered_pumps.append(
+                        HeatPumpDiscovery.HeatPumpInfo(
+                            uuid=pump.id,
+                            name=pump.name,
+                            model=model_string,
+                            sn=pump.serial_number,
+                            has_dhw=dhw,
+                        )
+                    )
         return discovered_pumps