Add support for ATEC operational statuses

klejejs · Dec 14, 2024 · 239b015 · 239b015
1 parent 68a6622
commit 239b015
Show file tree

Hide file tree

Showing 4 changed files with 1,529 additions and 0 deletions.
diff --git a/ThermiaOnlineAPI/const.py b/ThermiaOnlineAPI/const.py
@@ -70,6 +70,7 @@
     "REG_OPERATIONAL_STATUS_PRIO1"  # Operational status for most heat pumps
 )
 COMP_STATUS = "COMP_STATUS"  # Operational status for Diplomat heat pumps
+COMP_STATUS_ATEC = "COMP_STATUS_ATEC"  # Operational status for ATEC heat pumps
 COMP_STATUS_ITEC = "COMP_STATUS_ITEC"  # Operational status for ITEC heat pumps
 REG_OPERATIONAL_STATUS_PRIORITY_BITMASK = (
     "REG_OPERATIONAL_STATUS_PRIORITY_BITMASK"  # Operational status for Atlas heat pumps

diff --git a/ThermiaOnlineAPI/model/HeatPump.py b/ThermiaOnlineAPI/model/HeatPump.py
@@ -30,6 +30,7 @@
     REG_RETURN_LINE,
     COMP_POWER_STATUS,
     COMP_STATUS,
+    COMP_STATUS_ATEC,
     COMP_STATUS_ITEC,
     REG_SUPPLY_LINE,
     DATETIME_FORMAT,
@@ -380,6 +381,13 @@ def __get_operational_statuses_from_operational_status(self) -> Optional[Dict]:
             )
             return data.get("valueNames", [])
 
+        # Try to get the data from the COMP_STATUS_ATEC register
+        data = self.__get_register_from_operational_status(COMP_STATUS_ATEC)
+        if data is not None:
+            self.__device_config["operational_status_register"] = COMP_STATUS_ATEC
+            self.__device_config["operational_status_valueNamePrefix"] = "COMP_VALUE_"
+            return data.get("valueNames", [])
+
         # Try to get the data from the COMP_STATUS_ITEC register
         data = self.__get_register_from_operational_status(COMP_STATUS_ITEC)
         if data is not None: