23 files

custom_components/midea_ac_lan/switch.py

@@ -7,8 +7,11 @@ from homeassistant.const import CONF_DEVICE_ID, CONF_SWITCHES, Platform
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity import ToggleEntity
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from midealan.device import DeviceType
+from midealan.devices.ed import DeviceAttributes as EDAttributes
+from midealan.devices.ed import MideaEDDevice
 
-from .const import DEVICES, DOMAIN
+from .const import DEVICES, DOMAIN, supports_device
 from .midea_devices import MIDEA_DEVICES
 from .midea_entity import MideaEntity
 
@@ -27,9 +30,25 @@ async def async_setup_entry(
         "dict",
         MIDEA_DEVICES[device.device_type]["entities"],
     ).items():
-        if config["type"] == Platform.SWITCH and entity_key in extra_switches:
+        if (
+            config["type"] != Platform.SWITCH
+            or not supports_device(device.model, device.subtype, config)
+            or (not config.get("default") and entity_key not in extra_switches)
+        ):
+            continue
+        required_attribute = config.get("required_attribute")
+        if (
+            required_attribute is not None
+            and required_attribute not in device.attributes
+        ):
+            continue
+        if device.device_type == DeviceType.ED and entity_key == "tea_bar":
+            dev = MideaEDTeaBarBoilSwitch(device, entity_key)
+        elif device.device_type == DeviceType.ED and entity_key == "tea_bar_child_lock":
+            dev = MideaEDTeaBarChildLockSwitch(device, entity_key)
+        else:
             dev = MideaSwitch(device, entity_key)
-            switches.append(dev)
+        switches.append(dev)
     async_add_entities(switches)
 
 
@@ -48,3 +67,44 @@ class MideaSwitch(MideaEntity, ToggleEntity):
     def turn_off(self, **kwargs: Any) -> None:  # ruff:ignore[any-type, unused-method-argument]
         """Turn off switch."""
         self._device.set_attribute(attr=self._entity_key, value=False)
+
+
+class MideaEDTeaBarBoilSwitch(MideaSwitch):
+    """Accessible one-control switch for a subtype-395 tea bar."""
+
+    _device: MideaEDDevice
+
+    @property
+    def is_on(self) -> bool:
+        """Whether the automatic fill-and-boil cycle is active."""
+        return bool(
+            self._device.get_attribute(EDAttributes.boiling)
+            or self._device.get_attribute(EDAttributes.dispensing),
+        )
+
+    def turn_on(self, **kwargs: Any) -> None:  # ruff:ignore[any-type, unused-method-argument]
+        """Start the appliance's normal automatic fill-and-boil cycle."""
+        self._device.set_attribute(EDAttributes.boiling, True)
+
+    def turn_off(self, **kwargs: Any) -> None:  # ruff:ignore[any-type, unused-method-argument]
+        """Stop the active boil cycle."""
+        self._device.set_attribute(EDAttributes.boiling, False)
+
+
+class MideaEDTeaBarChildLockSwitch(MideaSwitch):
+    """Expose the tea-bar child lock with on/off switch semantics."""
+
+    _device: MideaEDDevice
+
+    @property
+    def is_on(self) -> bool:
+        """Whether the appliance child lock is enabled."""
+        return bool(self._device.get_attribute(EDAttributes.child_lock))
+
+    def turn_on(self, **kwargs: Any) -> None:  # ruff:ignore[any-type, unused-method-argument]
+        """Enable the child lock using the official model command."""
+        self._device.set_attribute(EDAttributes.child_lock, True)
+
+    def turn_off(self, **kwargs: Any) -> None:  # ruff:ignore[any-type, unused-method-argument]
+        """Disable the child lock using the official model command."""
+        self._device.set_attribute(EDAttributes.child_lock, False)