163 lines
4.8 KiB
Python
163 lines
4.8 KiB
Python
"""Support for the Airzone sensors."""
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Any, Final
|
|
|
|
from aioairzone.common import GrilleAngle, SleepTimeout
|
|
from aioairzone.const import (
|
|
API_COLD_ANGLE,
|
|
API_HEAT_ANGLE,
|
|
API_SLEEP,
|
|
AZD_COLD_ANGLE,
|
|
AZD_HEAT_ANGLE,
|
|
AZD_NAME,
|
|
AZD_SLEEP,
|
|
AZD_ZONES,
|
|
)
|
|
|
|
from homeassistant.components.select import SelectEntity, SelectEntityDescription
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import EntityCategory
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from .const import DOMAIN
|
|
from .coordinator import AirzoneUpdateCoordinator
|
|
from .entity import AirzoneEntity, AirzoneZoneEntity
|
|
|
|
|
|
@dataclass
|
|
class AirzoneSelectDescriptionMixin:
|
|
"""Define an entity description mixin for select entities."""
|
|
|
|
api_param: str
|
|
options_dict: dict[str, int]
|
|
|
|
|
|
@dataclass
|
|
class AirzoneSelectDescription(SelectEntityDescription, AirzoneSelectDescriptionMixin):
|
|
"""Class to describe an Airzone select entity."""
|
|
|
|
|
|
GRILLE_ANGLE_DICT: Final[dict[str, int]] = {
|
|
"90deg": GrilleAngle.DEG_90,
|
|
"50deg": GrilleAngle.DEG_50,
|
|
"45deg": GrilleAngle.DEG_45,
|
|
"40deg": GrilleAngle.DEG_40,
|
|
}
|
|
|
|
SLEEP_DICT: Final[dict[str, int]] = {
|
|
"off": SleepTimeout.SLEEP_OFF,
|
|
"30m": SleepTimeout.SLEEP_30,
|
|
"60m": SleepTimeout.SLEEP_60,
|
|
"90m": SleepTimeout.SLEEP_90,
|
|
}
|
|
|
|
|
|
ZONE_SELECT_TYPES: Final[tuple[AirzoneSelectDescription, ...]] = (
|
|
AirzoneSelectDescription(
|
|
api_param=API_COLD_ANGLE,
|
|
entity_category=EntityCategory.CONFIG,
|
|
key=AZD_COLD_ANGLE,
|
|
name="Cold Angle",
|
|
options=list(GRILLE_ANGLE_DICT),
|
|
options_dict=GRILLE_ANGLE_DICT,
|
|
translation_key="grille_angles",
|
|
),
|
|
AirzoneSelectDescription(
|
|
api_param=API_HEAT_ANGLE,
|
|
entity_category=EntityCategory.CONFIG,
|
|
key=AZD_HEAT_ANGLE,
|
|
name="Heat Angle",
|
|
options=list(GRILLE_ANGLE_DICT),
|
|
options_dict=GRILLE_ANGLE_DICT,
|
|
translation_key="grille_angles",
|
|
),
|
|
AirzoneSelectDescription(
|
|
api_param=API_SLEEP,
|
|
entity_category=EntityCategory.CONFIG,
|
|
key=AZD_SLEEP,
|
|
name="Sleep",
|
|
options=list(SLEEP_DICT),
|
|
options_dict=SLEEP_DICT,
|
|
translation_key="sleep_times",
|
|
),
|
|
)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
|
) -> None:
|
|
"""Add Airzone sensors from a config_entry."""
|
|
coordinator = hass.data[DOMAIN][entry.entry_id]
|
|
|
|
entities: list[AirzoneBaseSelect] = []
|
|
|
|
for system_zone_id, zone_data in coordinator.data[AZD_ZONES].items():
|
|
for description in ZONE_SELECT_TYPES:
|
|
if description.key in zone_data:
|
|
entities.append(
|
|
AirzoneZoneSelect(
|
|
coordinator,
|
|
description,
|
|
entry,
|
|
system_zone_id,
|
|
zone_data,
|
|
)
|
|
)
|
|
|
|
async_add_entities(entities)
|
|
|
|
|
|
class AirzoneBaseSelect(AirzoneEntity, SelectEntity):
|
|
"""Define an Airzone select."""
|
|
|
|
entity_description: AirzoneSelectDescription
|
|
values_dict: dict[int, str]
|
|
|
|
@callback
|
|
def _handle_coordinator_update(self) -> None:
|
|
"""Update attributes when the coordinator updates."""
|
|
self._async_update_attrs()
|
|
super()._handle_coordinator_update()
|
|
|
|
def _get_current_option(self) -> str | None:
|
|
value = self.get_airzone_value(self.entity_description.key)
|
|
return self.values_dict.get(value)
|
|
|
|
@callback
|
|
def _async_update_attrs(self) -> None:
|
|
"""Update select attributes."""
|
|
self._attr_current_option = self._get_current_option()
|
|
|
|
|
|
class AirzoneZoneSelect(AirzoneZoneEntity, AirzoneBaseSelect):
|
|
"""Define an Airzone Zone select."""
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: AirzoneUpdateCoordinator,
|
|
description: AirzoneSelectDescription,
|
|
entry: ConfigEntry,
|
|
system_zone_id: str,
|
|
zone_data: dict[str, Any],
|
|
) -> None:
|
|
"""Initialize."""
|
|
super().__init__(coordinator, entry, system_zone_id, zone_data)
|
|
|
|
self._attr_name = f"{zone_data[AZD_NAME]} {description.name}"
|
|
self._attr_unique_id = (
|
|
f"{self._attr_unique_id}_{system_zone_id}_{description.key}"
|
|
)
|
|
self.entity_description = description
|
|
self.values_dict = {v: k for k, v in description.options_dict.items()}
|
|
|
|
self._async_update_attrs()
|
|
|
|
async def async_select_option(self, option: str) -> None:
|
|
"""Change the selected option."""
|
|
param = self.entity_description.api_param
|
|
value = self.entity_description.options_dict[option]
|
|
await self._async_update_hvac_params({param: value})
|