Added Midea Dishwasher Integration

2026-06-12 11:51:37 -04:00
parent 8843bf77a5
commit 89ac3bd24f
47 changed files with 13150 additions and 245 deletions
@@ -0,0 +1,140 @@
+"""Midea Humidifier entries."""
+
+import logging
+from typing import Any, TypeAlias, cast
+
+from homeassistant.components.humidifier import (
+    HumidifierDeviceClass,
+    HumidifierEntity,
+    HumidifierEntityFeature,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_DEVICE_ID, CONF_SWITCHES, Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from midealocal.device import DeviceType
+from midealocal.devices.a1 import MideaA1Device
+from midealocal.devices.fd import MideaFDDevice
+
+from .const import DEVICES, DOMAIN
+from .midea_devices import MIDEA_DEVICES
+from .midea_entity import MideaEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up humidifier entries."""
+    device_id = config_entry.data.get(CONF_DEVICE_ID)
+    device = hass.data[DOMAIN][DEVICES].get(device_id)
+    extra_switches = config_entry.options.get(CONF_SWITCHES, [])
+    devs: list[MideaA1Humidifier | MideaFDHumidifier] = []
+    for entity_key, config in cast(
+        "dict",
+        MIDEA_DEVICES[device.device_type]["entities"],
+    ).items():
+        if config["type"] == Platform.HUMIDIFIER and (
+            config.get("default") or entity_key in extra_switches
+        ):
+            if device.device_type == DeviceType.A1:
+                devs.append(MideaA1Humidifier(device, entity_key))
+            if device.device_type == DeviceType.FD:
+                devs.append(MideaFDHumidifier(device, entity_key))
+    async_add_entities(devs)
+
+
+MideaHumidifierDevice: TypeAlias = MideaFDDevice | MideaA1Device
+
+
+class MideaHumidifier(MideaEntity, HumidifierEntity):
+    """Midea Humidifier Entries Base Class."""
+
+    _device: MideaHumidifierDevice
+
+    def __init__(self, device: MideaHumidifierDevice, entity_key: str) -> None:
+        """Midea Humidifier entity init."""
+        super().__init__(device, entity_key)
+
+    @property
+    def current_humidity(self) -> float | None:
+        """Midea Humidifier current humidity."""
+        return cast("float", self._device.get_attribute("current_humidity"))
+
+    @property
+    def target_humidity(self) -> float:
+        """Midea Humidifier target humidity."""
+        return cast("float", self._device.get_attribute("target_humidity"))
+
+    @property
+    def mode(self) -> str:
+        """Midea Humidifier mode."""
+        return cast("str", self._device.get_attribute("mode"))
+
+    @property
+    def available_modes(self) -> list[str] | None:
+        """Midea Humidifier available modes."""
+        return cast("list", self._device.modes)
+
+    def set_humidity(self, humidity: int) -> None:
+        """Midea Humidifier set humidity."""
+        self._device.set_attribute("target_humidity", humidity)
+
+    def set_mode(self, mode: str) -> None:
+        """Midea Humidifier set mode."""
+        self._device.set_attribute("mode", mode)
+
+    @property
+    def is_on(self) -> bool:
+        """Midea Humidifier is on."""
+        return cast("bool", self._device.get_attribute(attr="power"))
+
+    def turn_on(self, **kwargs: Any) -> None:  # noqa: ANN401, ARG002
+        """Midea Humidifier turn on."""
+        self._device.set_attribute(attr="power", value=True)
+
+    def turn_off(self, **kwargs: Any) -> None:  # noqa: ANN401, ARG002
+        """Midea Humidifier turn off."""
+        self._device.set_attribute(attr="power", value=False)
+
+    def update_state(self, status: Any) -> None:  # noqa: ANN401, ARG002
+        """Midea Humidifier update state."""
+        if not self.hass:
+            _LOGGER.warning(
+                "Humidifier update_state skipped for %s [%s]: HASS is None",
+                self.name,
+                type(self),
+            )
+            return
+        self.schedule_update_ha_state()
+
+
+class MideaA1Humidifier(MideaHumidifier):
+    """Midea A1 Humidifier Entries."""
+
+    _device: MideaA1Device
+
+    def __init__(self, device: MideaA1Device, entity_key: str) -> None:
+        """Midea A1 Humidifier entity init."""
+        super().__init__(device, entity_key)
+        self._attr_min_humidity: float = 35
+        self._attr_max_humidity: float = 85
+        self._attr_device_class = HumidifierDeviceClass.DEHUMIDIFIER
+        self._attr_supported_features = HumidifierEntityFeature.MODES
+
+
+class MideaFDHumidifier(MideaHumidifier):
+    """Midea FD Humidifier Entries."""
+
+    _device: MideaFDDevice
+
+    def __init__(self, device: MideaFDDevice, entity_key: str) -> None:
+        """Midea FD Humidifier entity init."""
+        super().__init__(device, entity_key)
+        self._attr_min_humidity: float = 35
+        self._attr_max_humidity: float = 85
+        self._attr_device_class = HumidifierDeviceClass.HUMIDIFIER
+        self._attr_supported_features = HumidifierEntityFeature.MODES