24 files

custom_components/midea_ac_lan/water_heater.py

@@ -22,16 +22,16 @@ from homeassistant.const import (
 )
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from midealocal.device import DeviceType
-from midealocal.devices.c3 import DeviceAttributes as C3Attributes
-from midealocal.devices.c3 import MideaC3Device
-from midealocal.devices.cd import DeviceAttributes as CDAttributes
-from midealocal.devices.cd import MideaCDDevice
-from midealocal.devices.e2 import DeviceAttributes as E2Attributes
-from midealocal.devices.e2 import MideaE2Device
-from midealocal.devices.e3 import MideaE3Device
-from midealocal.devices.e6 import DeviceAttributes as E6Attributes
-from midealocal.devices.e6 import MideaE6Device
+from midealan.device import DeviceType
+from midealan.devices.c3 import DeviceAttributes as C3Attributes
+from midealan.devices.c3 import MideaC3Device
+from midealan.devices.cd import DeviceAttributes as CDAttributes
+from midealan.devices.cd import MideaCDDevice
+from midealan.devices.e2 import DeviceAttributes as E2Attributes
+from midealan.devices.e2 import MideaE2Device
+from midealan.devices.e3 import MideaE3Device
+from midealan.devices.e6 import DeviceAttributes as E6Attributes
+from midealan.devices.e6 import MideaE6Device
 
 from .const import DEVICES, DOMAIN
 from .midea_devices import MIDEA_DEVICES
@@ -242,6 +242,16 @@ class MideaE3WaterHeater(MideaWaterHeater):
         """Midea E3 Water Heater entity init."""
         super().__init__(device, entity_key)
 
+    @property
+    def supported_features(self) -> WaterHeaterEntityFeature:
+        """Midea E3 Water Heater supported features."""
+        # E3 implements turn_on/turn_off and reports on/off state, so advertise
+        # ON_OFF to self-document that support (matches E2).
+        return (
+            WaterHeaterEntityFeature.TARGET_TEMPERATURE
+            | WaterHeaterEntityFeature.ON_OFF
+        )
+
     @property
     def min_temp(self) -> float:
         """Midea E3 Water Heater min temperature."""
@@ -276,6 +286,16 @@ class MideaC3WaterHeater(MideaWaterHeater):
         """Midea C3 Water Heater entity init."""
         super().__init__(device, entity_key)
 
+    @property
+    def supported_features(self) -> WaterHeaterEntityFeature:
+        """Midea C3 Water Heater supported features."""
+        # C3 implements turn_on/turn_off (dhw_power) and reports on/off state,
+        # so advertise ON_OFF to self-document that support (matches E2).
+        return (
+            WaterHeaterEntityFeature.TARGET_TEMPERATURE
+            | WaterHeaterEntityFeature.ON_OFF
+        )
+
     @property
     def current_operation(self) -> str:
         """Midea C3 Water Heater current operation."""
@@ -356,6 +376,16 @@ class MideaE6WaterHeater(MideaWaterHeater):
             self._use
         ]
 
+    @property
+    def supported_features(self) -> WaterHeaterEntityFeature:
+        """Midea E6 Water Heater supported features."""
+        # E6 implements turn_on/turn_off and reports on/off state, so advertise
+        # ON_OFF to self-document that support (matches E2).
+        return (
+            WaterHeaterEntityFeature.TARGET_TEMPERATURE
+            | WaterHeaterEntityFeature.ON_OFF
+        )
+
     @property
     def current_operation(self) -> str:
         """Midea E6 Water Heater current operation."""