Fix set hvac mode for BSB devices (#390)

fustom · Jan 10, 2025 · f7e9acb · f7e9acb
1 parent 41fbfd1
commit f7e9acb
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/custom_components/ariston/climate.py b/custom_components/ariston/climate.py
@@ -4,7 +4,7 @@
 
 import logging
 
-from ariston.const import PlantMode, ZoneMode
+from ariston.const import PlantMode, ZoneMode, BsbZoneMode
 from homeassistant.components.climate import (
     ClimateEntity,
     ClimateEntityFeature,
@@ -247,8 +247,13 @@ async def async_set_hvac_mode(self, hvac_mode):
                     )
                 else:
                     await self.device.async_set_zone_mode(ZoneMode.MANUAL, self.zone)
-        else:
-            await self.device.async_set_zone_mode(hvac_mode, self.zone)
+        # Plant mode is not supported (BSB device)
+        elif hvac_mode == HVACMode.OFF:
+            await self.device.async_set_zone_mode(BsbZoneMode.OFF, self.zone)
+        elif hvac_mode == HVACMode.AUTO:
+            await self.device.async_set_zone_mode(BsbZoneMode.TIME_PROGRAM, self.zone)
+        elif hvac_mode == HVACMode.HEAT:
+            await self.device.async_set_zone_mode(BsbZoneMode.MANUAL, self.zone)
 
         self.async_write_ha_state()