Fix space

homeassistant/components/zha/climate.py

@@ -823,3 +823,153 @@ class ZONNSMARTThermostat(Thermostat):
             return await self._thrm.write_attributes_safe(
                 {"operation_preset": 4}, manufacturer=mfg_code
             )
+
+
+@MULTI_MATCH(
+    cluster_handler_names={"legrand_wire_pilot_cluster"},
+)
+class LegrandWirePilotThermostat(ZhaEntity, ClimateEntity):
+    """Legrand wire pilot Thermostat implementation."""
+
+    PRESET_COMFORT_MINUS_1 = "comfort_minus_1"
+    PRESET_COMFORT_MINUS_2 = "comfort_minus_2"
+    PRESET_FROST_PROTECTION = "frost_protection"
+    PRESET_OFF = "off"
+
+    _attr_temperature_unit = UnitOfTemperature.CELSIUS
+    _attr_translation_key: str = "legrand_thermostat"
+    _enable_turn_on_off_backwards_compatibility = False
+
+    _heat_mode = int | None
+
+    def __init__(self, unique_id, zha_device, cluster_handlers, **kwargs):
+        """Initialize ZHA Thermostat instance."""
+        super().__init__(unique_id, zha_device, cluster_handlers, **kwargs)
+        self._wire_pilot_cluster = self.cluster_handlers.get(
+            "legrand_wire_pilot_cluster"
+        )
+
+    async def async_added_to_hass(self) -> None:
+        """Run when about to be added to hass."""
+
+        await super().async_added_to_hass()
+
+        self._heat_mode = self._wire_pilot_cluster.cluster.get("heat_mode")
+        self.async_accept_signal(
+            self._wire_pilot_cluster, SIGNAL_ATTR_UPDATED, self.async_attribute_updated
+        )
+
+    async def async_attribute_updated(self, attr_id, attr_name, value):
+        """Handle attribute update from device."""
+
+        if attr_name == "heat_mode":
+            self._heat_mode = value
+        self.async_write_ha_state()
+
+    @property
+    def hvac_mode(self) -> HVACMode | None:
+        """Return HVAC operation mode."""
+
+        heat_mode = self._heat_mode
+        if not isinstance(heat_mode, int):
+            return None
+        if heat_mode == 5:
+            return HVACMode.OFF
+        if heat_mode >= 0:
+            return HVACMode.HEAT
+        return None
+
+    @property
+    def hvac_modes(self):
+        """Return the list of available HVAC operation modes."""
+        return [HVACMode.HEAT, HVACMode.OFF]
+
+    @property
+    def preset_mode(self):
+        """Return current preset mode."""
+
+        heat_mode = self._heat_mode
+        if not isinstance(heat_mode, int):
+            return None
+        if heat_mode == 0:
+            return PRESET_COMFORT
+        if heat_mode == 1:
+            return self.PRESET_COMFORT_MINUS_1
+        if heat_mode == 2:
+            return self.PRESET_COMFORT_MINUS_2
+        if heat_mode == 3:
+            return PRESET_ECO
+        if heat_mode == 4:
+            return self.PRESET_FROST_PROTECTION
+        return None
+
+    @property
+    def preset_modes(self) -> list[str] | None:
+        """Return supported preset modes."""
+
+        return [
+            PRESET_COMFORT,
+            self.PRESET_COMFORT_MINUS_1,
+            self.PRESET_COMFORT_MINUS_2,
+            PRESET_ECO,
+            self.PRESET_FROST_PROTECTION,
+            PRESET_NONE,
+        ]
+
+    @property
+    def supported_features(self) -> ClimateEntityFeature:
+        """Return the list of supported features."""
+
+        return (
+            ClimateEntityFeature.PRESET_MODE
+            | ClimateEntityFeature.TURN_OFF
+            | ClimateEntityFeature.TURN_ON
+        )
+
+    async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
+        """Set new target operation mode."""
+        if hvac_mode not in self.hvac_modes:
+            self.warning(
+                "can't set '%s' mode. Supported modes are: %s",
+                hvac_mode,
+                self.hvac_modes,
+            )
+            return
+
+        if hvac_mode == self.hvac_mode:
+            return
+
+        heat_mode = 5 if hvac_mode == HVACMode.OFF else 0
+        mfg_code = self._zha_device.manufacturer_code
+        await self._wire_pilot_cluster.write_attributes_safe(
+            {"heat_mode": heat_mode}, manufacturer=mfg_code
+        )
+
+    async def async_set_preset_mode(self, preset_mode: str) -> None:
+        """Set new preset mode."""
+        if self.preset_modes and (preset_mode not in self.preset_modes):
+            self.debug("Preset mode '%s' is not supported", preset_mode)
+            return
+
+        heat_mode: int | None = None
+
+        if preset_mode == PRESET_COMFORT:
+            heat_mode = 0
+        elif preset_mode == self.PRESET_COMFORT_MINUS_1:
+            heat_mode = 1
+        elif preset_mode == self.PRESET_COMFORT_MINUS_2:
+            heat_mode = 2
+        elif preset_mode == PRESET_ECO:
+            heat_mode = 3
+        elif preset_mode == self.PRESET_FROST_PROTECTION:
+            heat_mode = 4
+        elif preset_mode == PRESET_NONE:
+            heat_mode = 5
+
+        if heat_mode is None:
+            return
+
+        mfg_code = self._zha_device.manufacturer_code
+        await self._wire_pilot_cluster.write_attributes_safe(
+            {"heat_mode": heat_mode}, manufacturer=mfg_code
+        )

homeassistant/components/zha/icons.json

@@ -5,6 +5,20 @@
         "default": "mdi:hand-wave"
       }
     },
+    "climate": {
+      "legrand_thermostat": {
+        "state_attributes": {
+          "preset_mode": {
+            "state": {
+              "comfort_minus_1": "mdi:numeric-1-circle",
+              "comfort_minus_2": "mdi:numeric-2-circle",
+              "frost_protection": "mdi:snowflake-thermometer",
+              "none": "mdi:power"
+            }
+          }
+        }
+      }
+    },
     "number": {
       "timer_duration": {
         "default": "mdi:timer"

homeassistant/components/zha/strings.json

@@ -588,6 +588,18 @@
     "climate": {
       "thermostat": {
         "name": "[%key:component::climate::entity_component::_::name%]"
+      },
+      "legrand_thermostat": {
+        "name": "[%key:component::climate::entity_component::_::name%]",
+        "state_attributes": {
+          "preset_mode": {
+            "state": {
+              "comfort_minus_1": "Comfort -1 °C",
+              "comfort_minus_2": "Comfort -2 °C",
+              "frost_protection": "Frost protection"
+            }
+          }
+        }
       }
     },
     "cover": {
@@ -991,6 +1003,9 @@
       },
       "buzzer_manual_alarm": {
         "name": "Buzzer manual alarm"
+      },
+      "wire_pilot_mode": {
+        "name": "Wire pilot mode"
       }
     }
   }