144 lines
4.5 KiB
Python
144 lines
4.5 KiB
Python
"""Adds fan entity for each dehumidifer appliance."""
|
|
|
|
import logging
|
|
from typing import Any, Final
|
|
|
|
from homeassistant.components.fan import (
|
|
FanEntityFeature,
|
|
FanEntity,
|
|
)
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from custom_components.midea_dehumidifier_lan.const import ATTR_FAN_SPEED, DOMAIN
|
|
from custom_components.midea_dehumidifier_lan.appliance_coordinator import (
|
|
ApplianceEntity,
|
|
ApplianceUpdateCoordinator,
|
|
)
|
|
from custom_components.midea_dehumidifier_lan.hub import Hub
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
MODE_NONE: Final = "None"
|
|
MODE_AUTO: Final = "Auto"
|
|
MODE_LOW: Final = "Low"
|
|
MODE_MEDIUM: Final = "Medium"
|
|
MODE_HIGH: Final = "High"
|
|
|
|
PRESET_MODES_7: Final = [MODE_LOW, MODE_MEDIUM, MODE_HIGH]
|
|
PRESET_MODES_3: Final = [MODE_LOW, MODE_HIGH]
|
|
PRESET_MODES_2: Final = [MODE_AUTO]
|
|
|
|
_FAN_SPEEDS = {2: PRESET_MODES_2, 3: PRESET_MODES_3, 7: PRESET_MODES_7}
|
|
_ON_SPEED = {2: MODE_AUTO, 3: MODE_HIGH, 7: MODE_HIGH}
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Sets up fan entity for dehumidifer"""
|
|
|
|
hub: Hub = hass.data[DOMAIN][config_entry.entry_id]
|
|
|
|
async_add_entities(
|
|
DehumidiferFan(c) for c in hub.coordinators if c.is_dehumidifier()
|
|
)
|
|
|
|
|
|
# pylint: disable=too-many-ancestors
|
|
class DehumidiferFan(ApplianceEntity, FanEntity):
|
|
"""Entity for managing dehumidifer fan"""
|
|
|
|
_attr_supported_features = (
|
|
FanEntityFeature.PRESET_MODE |
|
|
FanEntityFeature.TURN_OFF |
|
|
FanEntityFeature.TURN_ON
|
|
)
|
|
|
|
_attr_preset_modes = PRESET_MODES_7
|
|
_attr_speed_count = len(PRESET_MODES_7)
|
|
_name_suffix = " Fan"
|
|
_on_speed = MODE_MEDIUM
|
|
|
|
def __init__(self, coordinator: ApplianceUpdateCoordinator) -> None:
|
|
|
|
super().__init__(coordinator)
|
|
self._fan_speeds = {
|
|
MODE_NONE: 0,
|
|
MODE_LOW: 40,
|
|
MODE_MEDIUM: 60,
|
|
MODE_HIGH: 80,
|
|
MODE_AUTO: 101,
|
|
}
|
|
|
|
@property
|
|
def is_on(self):
|
|
# Override parent logic
|
|
return self._attr_is_on
|
|
|
|
def on_online(self, update: bool) -> None:
|
|
supports = self.dehumidifier().capabilities
|
|
fan_capability = supports.get("fan_speed", 0)
|
|
self._attr_preset_modes = _FAN_SPEEDS.get(fan_capability, PRESET_MODES_7)
|
|
self._on_speed = _ON_SPEED.get(fan_capability, MODE_HIGH)
|
|
self._attr_speed_count = len(self._attr_preset_modes)
|
|
return super().on_online(update)
|
|
|
|
def on_update(self) -> None:
|
|
fan_speed = self.dehumidifier().fan_speed
|
|
self._attr_percentage = fan_speed
|
|
self._attr_is_on = fan_speed > self._fan_speeds[MODE_LOW]
|
|
for mode, mode_speed in self._fan_speeds.items():
|
|
if fan_speed <= mode_speed:
|
|
self._attr_preset_mode = mode
|
|
break
|
|
else:
|
|
self._attr_preset_mode = MODE_NONE
|
|
|
|
def set_preset_mode(self, preset_mode: str) -> None:
|
|
"""Set the preset mode of the fan."""
|
|
speed = self._fan_speeds.get(preset_mode, None)
|
|
_LOGGER.debug("Setting speed to %s", speed)
|
|
if speed is not None:
|
|
self.apply(ATTR_FAN_SPEED, speed)
|
|
else:
|
|
_LOGGER.warning("Unsupported fan mode %s", preset_mode)
|
|
|
|
def set_percentage(self, percentage: int) -> None:
|
|
"""Set the speed percentage of the fan."""
|
|
_LOGGER.debug("Setting percentage to %s", percentage)
|
|
|
|
self.apply(ATTR_FAN_SPEED, percentage)
|
|
|
|
def turn_on(
|
|
self,
|
|
speed: str = None,
|
|
percentage: int = None,
|
|
preset_mode: str = None,
|
|
**kwargs,
|
|
) -> None:
|
|
"""Turns fan to medium speed."""
|
|
updated = False
|
|
if preset_mode is not None:
|
|
self.set_preset_mode(preset_mode)
|
|
updated = True
|
|
if percentage is not None:
|
|
self.set_percentage(percentage)
|
|
updated = True
|
|
if speed is not None:
|
|
self.set_speed(speed)
|
|
updated = True
|
|
# _LOGGER.debug("turn_on percentage=%s was_updated=%s", self._attr_percentage, updated)
|
|
if (
|
|
not updated
|
|
and (self._attr_percentage or 0) < self._fan_speeds[self._on_speed]
|
|
):
|
|
self.set_preset_mode(self._on_speed)
|
|
|
|
def turn_off(self, **kwargs: Any) -> None:
|
|
"""Turns fan to silent speed."""
|
|
self.set_preset_mode(MODE_LOW)
|