Initial commit.

2025-04-21 21:03:30 -04:00
commit d4f58c985f
2077 changed files with 75132 additions and 0 deletions
@@ -0,0 +1,152 @@
+"""Platform to locally control Tuya-based button devices."""
+
+import logging
+from functools import partial
+from .config_flow import col_to_select
+from homeassistant.helpers.selector import ObjectSelector
+
+import voluptuous as vol
+from homeassistant.const import CONF_DEVICE_CLASS
+from homeassistant.components.humidifier import (
+    DOMAIN,
+    HumidifierDeviceClass,
+    DEVICE_CLASSES_SCHEMA,
+    HumidifierEntity,
+    HumidifierEntityDescription,
+    HumidifierEntityFeature,
+)
+from homeassistant.components.humidifier.const import (
+    ATTR_MAX_HUMIDITY,
+    ATTR_MIN_HUMIDITY,
+    DEFAULT_MAX_HUMIDITY,
+    DEFAULT_MIN_HUMIDITY,
+)
+
+from .const import (
+    CONF_HUMIDIFIER_SET_HUMIDITY_DP,
+    CONF_HUMIDIFIER_CURRENT_HUMIDITY_DP,
+    CONF_HUMIDIFIER_MODE_DP,
+    CONF_HUMIDIFIER_AVAILABLE_MODES,
+    DictSelector,
+)
+
+from .entity import LocalTuyaEntity, async_setup_entry
+
+
+_LOGGER = logging.getLogger(__name__)
+
+
+def flow_schema(dps):
+    """Return schema used in config flow."""
+    return {
+        vol.Optional(CONF_HUMIDIFIER_SET_HUMIDITY_DP): col_to_select(dps, is_dps=True),
+        vol.Optional(CONF_HUMIDIFIER_CURRENT_HUMIDITY_DP): col_to_select(
+            dps, is_dps=True
+        ),
+        vol.Optional(CONF_HUMIDIFIER_MODE_DP): col_to_select(dps, is_dps=True),
+        vol.Required(ATTR_MIN_HUMIDITY, default=DEFAULT_MIN_HUMIDITY): int,
+        vol.Required(ATTR_MAX_HUMIDITY, default=DEFAULT_MAX_HUMIDITY): int,
+        vol.Optional(CONF_HUMIDIFIER_AVAILABLE_MODES, default={}): ObjectSelector(),
+        vol.Optional(CONF_DEVICE_CLASS): DEVICE_CLASSES_SCHEMA,
+    }
+
+
+class LocalTuyaHumidifier(LocalTuyaEntity, HumidifierEntity):
+    """Representation of a Localtuya Humidifier."""
+
+    _dp_mode = CONF_HUMIDIFIER_MODE_DP
+    _available_modes = CONF_HUMIDIFIER_AVAILABLE_MODES
+    _dp_current_humidity = CONF_HUMIDIFIER_CURRENT_HUMIDITY_DP
+    _dp_set_humidity = CONF_HUMIDIFIER_SET_HUMIDITY_DP
+    _mode_name_to_value = {}
+
+    def __init__(
+        self,
+        device,
+        config_entry,
+        humidifierID,
+        **kwargs,
+    ):
+        """Initialize the Tuya button."""
+        super().__init__(device, config_entry, humidifierID, _LOGGER, **kwargs)
+        self._state = None
+        self._current_mode = None
+
+        if self._config.get(self._dp_mode) and (
+            modes := self._config.get(self._available_modes, {})
+        ):
+            self._attr_supported_features |= HumidifierEntityFeature.MODES
+            modes = {
+                k: v if k else v.replace("_", " ").capitalize()
+                for k, v in modes.copy().items()
+            }
+        self._available_modes = DictSelector(modes)
+
+        self._attr_min_humidity = self._config.get(
+            ATTR_MIN_HUMIDITY, DEFAULT_MIN_HUMIDITY
+        )
+        self._attr_max_humidity = self._config.get(
+            ATTR_MAX_HUMIDITY, DEFAULT_MAX_HUMIDITY
+        )
+
+    @property
+    def is_on(self) -> bool:
+        """Return the device is on or off."""
+        return self._state
+
+    @property
+    def mode(self) -> str | None:
+        """Return the current mode."""
+        return self._current_mode
+
+    @property
+    def target_humidity(self) -> int | None:
+        """Return the humidity we try to reach."""
+        target_dp = self._config.get(self._dp_set_humidity, None)
+        return self.dp_value(target_dp) if target_dp else None
+
+    @property
+    def current_humidity(self) -> int | None:
+        """Return the current humidity."""
+        curr_humidity = self._config.get(self._dp_current_humidity)
+
+        return self.dp_value(self._dp_current_humidity) if curr_humidity else None
+
+    async def async_turn_on(self, **kwargs):
+        """Turn the device on."""
+        await self._device.set_dp(True, self._dp_id)
+
+    async def async_turn_off(self, **kwargs):
+        """Turn the device off."""
+        await self._device.set_dp(False, self._dp_id)
+
+    async def async_set_humidity(self, humidity: int) -> None:
+        """Set new target humidity."""
+        set_humidity_dp = self._config.get(self._dp_set_humidity, None)
+        if set_humidity_dp is None:
+            return None
+
+        await self._device.set_dp(humidity, set_humidity_dp)
+
+    @property
+    def available_modes(self):
+        """Return the list of presets that this device supports."""
+        return self._available_modes.names
+
+    async def async_set_mode(self, mode):
+        """Set new target preset mode."""
+        set_mode_dp = self._config.get(self._dp_mode, None)
+        if set_mode_dp is None:
+            return None
+
+        await self._device.set_dp(self._available_modes.to_tuya(mode), set_mode_dp)
+
+    def status_updated(self):
+        """Device status was updated."""
+        super().status_updated()
+        current_mode = self.dp_value(self._dp_mode)
+
+        self._current_mode = self._available_modes.to_ha(current_mode, "unknown")
+
+
+async_setup_entry = partial(async_setup_entry, DOMAIN, LocalTuyaHumidifier, flow_schema)