Add preset mode to LWZ

pail23 · Nov 24, 2023 · b77b165 · b77b165
1 parent ecceb72
commit b77b165
Show file tree

Hide file tree

Showing 4 changed files with 51 additions and 3 deletions.
diff --git a/custom_components/stiebel_eltron_isg/climate.py b/custom_components/stiebel_eltron_isg/climate.py
@@ -53,6 +53,9 @@
 PRESET_WATER_HEATING = "water_heating"
 PRESET_EMERGENCY = "emergency"
 PRESET_READY = "ready"
+PRESET_MANUAL = "manual"
+PRESET_AUTO = "auto"
+
 
 WPM_TO_HA_PRESET = {
     1: PRESET_READY,
@@ -93,6 +96,26 @@
     HVACMode.HEAT: 14,
 }
 
+LWZ_TO_HA_PRESET = {
+    1: PRESET_READY,
+    3: PRESET_COMFORT,
+    4: PRESET_ECO,
+    5: PRESET_WATER_HEATING,
+    11: PRESET_AUTO,
+    14: PRESET_MANUAL,
+    0: PRESET_EMERGENCY,
+}
+
+HA_TO_LWZ_PRESET = {
+    PRESET_READY: 1,
+    PRESET_COMFORT: 3,
+    PRESET_ECO: 4,
+    PRESET_WATER_HEATING: 5,
+    PRESET_AUTO: 11,
+    PRESET_MANUAL: 14,
+    PRESET_EMERGENCY: 0,
+}
+
 LWZ_TO_HA_FAN = {0: FAN_OFF, 1: FAN_LOW, 2: FAN_MEDIUM, 3: FAN_HIGH}
 HA_TO_LWZ_FAN = {k: i for i, k in LWZ_TO_HA_FAN.items()}
 
@@ -242,6 +265,15 @@ class StiebelEltronLWZClimateEntity(StiebelEltronISGClimateEntity):
     def __init__(self, coordinator, config_entry, description):
         """Initialize the climate entity."""
         self._attr_hvac_modes = [HVACMode.AUTO, HVACMode.OFF, HVACMode.HEAT]
+        self._attr_preset_modes = [
+            PRESET_READY,
+            PRESET_AUTO,
+            PRESET_MANUAL,
+            PRESET_ECO,
+            PRESET_COMFORT,
+            PRESET_WATER_HEATING,
+            PRESET_EMERGENCY,
+        ]
         self._attr_fan_modes = [FAN_OFF, FAN_LOW, FAN_MEDIUM, FAN_HIGH]
         super().__init__(coordinator, config_entry, description)
         self._attr_supported_features = (
@@ -258,6 +290,16 @@ def set_hvac_mode(self, hvac_mode: HVACMode) -> None:
         new_mode = HA_TO_LWZ_HVAC.get(hvac_mode)
         self.coordinator.set_data(OPERATION_MODE, new_mode)
 
+    @property
+    def preset_mode(self) -> str | None:
+        """Return current preset mode."""
+        return LWZ_TO_HA_PRESET.get(self.coordinator.data.get(OPERATION_MODE))
+
+    def set_preset_mode(self, preset_mode):
+        """Set new target preset mode."""
+        new_mode = HA_TO_LWZ_PRESET.get(preset_mode)
+        self.coordinator.set_data(OPERATION_MODE, new_mode)
+
     @property
     def fan_mode(self) -> str | None:
         """Return the fan setting. Requires ClimateEntityFeature.FAN_MODE."""

diff --git a/custom_components/stiebel_eltron_isg/strings.json b/custom_components/stiebel_eltron_isg/strings.json
@@ -27,7 +27,9 @@
                             "water_heating": "Water heating",
                             "emergency": "Emergency",
                             "program": "Program",
-                            "ready": "Ready"
+                            "ready": "Ready",
+                            "auto": "Auto",
+                            "manual": "Manual"
                         }
                     }
                 }

diff --git a/custom_components/stiebel_eltron_isg/translations/de.json b/custom_components/stiebel_eltron_isg/translations/de.json
@@ -27,7 +27,9 @@
                             "water_heating": "Warmwasserbetrieb",
                             "emergency": "Notbetrieb",
                             "program": "Programmbetrieb",
-                            "ready": "Bereitschaftsbetrieb"
+                            "ready": "Bereitschaftsbetrieb",
+                            "auto": "Autobetrieb",
+                            "manual": "Handbetrieb"
                         }
                     }
                 }

diff --git a/custom_components/stiebel_eltron_isg/translations/en.json b/custom_components/stiebel_eltron_isg/translations/en.json
@@ -27,7 +27,9 @@
                             "water_heating": "Water heating",
                             "emergency": "Emergency",
                             "program": "Program",
-                            "ready": "Ready"
+                            "ready": "Ready",
+                            "auto": "Auto",
+                            "manual": "Manual"
                         }
                     }
                 }