select.py

Go to the documentation of this file.

"""The Aprilaire select component."""
 
from __future__ import annotations
 
from collections.abc import Awaitable, Callable
from dataclasses import dataclass
from typing import cast
 
from pyaprilaire.const import Attribute
 
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .coordinator import AprilaireConfigEntry, AprilaireCoordinator
from .entity import BaseAprilaireEntity
 
AIR_CLEANING_EVENT_MAP = {0: "off", 3: "event_clean", 4: "allergies"}
AIR_CLEANING_MODE_MAP = {0: "off", 1: "constant_clean", 2: "automatic"}
FRESH_AIR_EVENT_MAP = {0: "off", 2: "3hour", 3: "24hour"}
FRESH_AIR_MODE_MAP = {0: "off", 1: "automatic"}
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: AprilaireConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up Aprilaire select devices."""
 
    coordinator = config_entry.runtime_data
 
    assert config_entry.unique_id is not None
 
    descriptions: list[AprilaireSelectDescription] = []
 
    if coordinator.data.get(Attribute.AIR_CLEANING_AVAILABLE) == 1:
        descriptions.extend(
            [
                AprilaireSelectDescription(
                    key="air_cleaning_event",
                    translation_key="air_cleaning_event",
                    options_map=AIR_CLEANING_EVENT_MAP,
                    event_value_key=Attribute.AIR_CLEANING_EVENT,
                    mode_value_key=Attribute.AIR_CLEANING_MODE,
                    is_event=True,
                    select_option_fn=coordinator.client.set_air_cleaning,
                ),
                AprilaireSelectDescription(
                    key="air_cleaning_mode",
                    translation_key="air_cleaning_mode",
                    options_map=AIR_CLEANING_MODE_MAP,
                    event_value_key=Attribute.AIR_CLEANING_EVENT,
                    mode_value_key=Attribute.AIR_CLEANING_MODE,
                    is_event=False,
                    select_option_fn=coordinator.client.set_air_cleaning,
                ),
            ]
        )
 
    if coordinator.data.get(Attribute.VENTILATION_AVAILABLE) == 1:
        descriptions.extend(
            [
                AprilaireSelectDescription(
                    key="fresh_air_event",
                    translation_key="fresh_air_event",
                    options_map=FRESH_AIR_EVENT_MAP,
                    event_value_key=Attribute.FRESH_AIR_EVENT,
                    mode_value_key=Attribute.FRESH_AIR_MODE,
                    is_event=True,
                    select_option_fn=coordinator.client.set_fresh_air,
                ),
                AprilaireSelectDescription(
                    key="fresh_air_mode",
                    translation_key="fresh_air_mode",
                    options_map=FRESH_AIR_MODE_MAP,
                    event_value_key=Attribute.FRESH_AIR_EVENT,
                    mode_value_key=Attribute.FRESH_AIR_MODE,
                    is_event=False,
                    select_option_fn=coordinator.client.set_fresh_air,
                ),
            ]
        )
 
    async_add_entities(
        AprilaireSelectEntity(coordinator, description, config_entry.unique_id)
        for description in descriptions
    )
 
 
@dataclass(frozen=True, kw_only=True)
class AprilaireSelectDescription(SelectEntityDescription):
    """Class describing Aprilaire select entities."""
 
    options_map: dict[int, str]
    event_value_key: str
    mode_value_key: str
    is_event: bool
    select_option_fn: Callable[[int, int], Awaitable]
 
 
class AprilaireSelectEntity(BaseAprilaireEntity, SelectEntity):
    """Base select entity for Aprilaire."""
 
    entity_description: AprilaireSelectDescription
 
    def __init__(
        self,
        coordinator: AprilaireCoordinator,
        description: AprilaireSelectDescription,
        unique_id: str,
    ) -> None:
        """Initialize a select for an Aprilaire device."""
 
        self.entity_descriptionentity_description = description
        self.values_mapvalues_map = {v: k for k, v in description.options_map.items()}
 
        super().__init__(coordinator, unique_id)
 
        self._attr_options_attr_options = list(description.options_map.values())
 
    @property
    def current_option(self) -> str:
        """Get the current option."""
 
        if self.entity_descriptionentity_description.is_event:
            value_key = self.entity_descriptionentity_description.event_value_key
        else:
            value_key = self.entity_descriptionentity_description.mode_value_key
 
        current_value = int(self.coordinator.data.get(value_key, 0))
 
        return self.entity_descriptionentity_description.options_map.get(current_value, "off")
 
    async def async_select_option(self, option: str) -> None:
        """Set the current option."""
 
        if self.entity_descriptionentity_description.is_event:
            event_value = self.values_mapvalues_map[option]
 
            mode_value = cast(
                int, self.coordinator.data.get(self.entity_descriptionentity_description.mode_value_key)
            )
        else:
            mode_value = self.values_mapvalues_map[option]
 
            event_value = cast(
                int, self.coordinator.data.get(self.entity_descriptionentity_description.event_value_key)
            )
 
        await self.entity_descriptionentity_description.select_option_fn(mode_value, event_value)