feat: add support for Ariston Lydos Wifi's boost mode (#163)

* feat: add support for Ariston Lydos Wifi's boost mode

This should address [this issue](fustom/ariston-remotethermo-home-assistant-v3#364) from the Home Assistant integration.

I've locally tested this and it works on my device.

* Organize AristonLydosDevice import

---------

Co-authored-by: fustom <[email protected]>

Author: ivan-gj

ariston/__init__.py

@@ -23,20 +23,22 @@
 from .nuos_split_device import AristonNuosSplitDevice
 from .base_device import AristonBaseDevice
 from .velis_base_device import AristonVelisBaseDevice
+from .lydos_device import AristonLydosDevice
 
 _LOGGER = logging.getLogger(__name__)
 
 _MAP_WHE_TYPES_TO_CLASS: dict[int, Type[AristonVelisBaseDevice]] = {
     WheType.Evo.value: AristonEvoOneDevice,
     WheType.LydosHybrid.value: AristonLydosHybridDevice,
-    WheType.Lydos.value: AristonEvoDevice,
+    WheType.Lydos.value: AristonLydosDevice,
     WheType.NuosSplit.value: AristonNuosSplitDevice,
     WheType.Andris2.value: AristonEvoDevice,
     WheType.Evo2.value: AristonEvoDevice,
     WheType.Lux2.value: AristonLux2Device,
     WheType.Lux.value: AristonLuxDevice,
 }
 
+
 class Ariston:
     """Ariston class"""
 

ariston/lydos_device.py

@@ -0,0 +1,39 @@
+"""Evo device class for Ariston module."""
+
+from __future__ import annotations
+
+import logging
+
+from .const import (
+    EvoDeviceProperties,
+    LuxPlantMode,
+    WaterHeaterMode,
+)
+from .evo_device import AristonEvoDevice
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class AristonLydosDevice(AristonEvoDevice):
+    """Class representing a physical Lydos Wi-Fi device, it's state and properties."""
+
+    @property
+    def water_heater_mode(self) -> type[WaterHeaterMode]:
+        """Return the water heater mode class"""
+        return LuxPlantMode
+
+    def set_water_heater_operation_mode(self, operation_mode: str):
+        """Set water heater operation mode"""
+        self.api.set_evo_mode(self.gw, self.water_heater_mode[operation_mode])
+        self.data[EvoDeviceProperties.MODE] = self.water_heater_mode[
+            operation_mode
+        ].value
+
+    async def async_set_water_heater_operation_mode(self, operation_mode: str):
+        """Async set water heater operation mode"""
+        await self.api.async_set_evo_mode(
+            self.gw, self.water_heater_mode[operation_mode]
+        )
+        self.data[EvoDeviceProperties.MODE] = self.water_heater_mode[
+            operation_mode
+        ].value