Make more standard set comfocollmode (michaelarnauts#27)

szibis · May 4, 2024 · a07a4b9 · a07a4b9
1 parent bd27eeb
commit a07a4b9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/aiocomfoconnect/comfoconnect.py b/aiocomfoconnect/comfoconnect.py
@@ -27,6 +27,7 @@
     VentilationSetting,
     VentilationSpeed,
     VentilationTemperatureProfile,
+    ComfoCoolMode,
 )
 from aiocomfoconnect.exceptions import (
     AioComfoConnectNotConnected,
@@ -397,9 +398,9 @@ async def get_comfocool_mode(self):
 
     async def set_comfocool_mode(self, mode: Literal["auto", "off"], timeout=-1):
         """Set the comfocool mode (auto / off)."""
-        if mode == "auto":
+        if mode == ComfoCoolMode.AUTO:
             await self.cmd_rmi_request(bytes([0x85, UNIT_SCHEDULE, SUBUNIT_05, 0x01]))
-        else:
+        elif mode == ComfoCoolMode.OFF:
             await self.cmd_rmi_request(bytestring([0x84, UNIT_SCHEDULE, SUBUNIT_05, 0x01, 0x00, 0x00, 0x00, 0x00, timeout.to_bytes(4, "little", signed=True), 0x00]))
 
     async def get_temperature_profile(self):

diff --git a/aiocomfoconnect/const.py b/aiocomfoconnect/const.py
@@ -207,3 +207,9 @@ class VentilationSpeed:
     LOW = "low"
     MEDIUM = "medium"
     HIGH = "high"
+
+class ComfoCoolMode:
+    """Enum for ventilation settings."""
+
+    AUTO = "auto"
+    OFF = "off"