mirror of
https://github.com/home-assistant/core.git
synced 2026-06-11 11:41:42 +02:00
David Bonnes
117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
"""Support for Button entities of the Evohome integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import evohomeasync2 as evo
|
|
|
|
from homeassistant.components.button import ButtonEntity
|
|
from homeassistant.const import EntityCategory
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
from .const import EVOHOME_DATA
|
|
from .coordinator import EvoDataUpdateCoordinator
|
|
from .entity import EvoEntity, is_valid_zone
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass: HomeAssistant,
|
|
_: ConfigType,
|
|
async_add_entities: AddEntitiesCallback,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> None:
|
|
"""Set up the button platform for Evohome."""
|
|
|
|
if discovery_info is None:
|
|
return
|
|
|
|
coordinator = hass.data[EVOHOME_DATA].coordinator
|
|
tcs = hass.data[EVOHOME_DATA].tcs
|
|
|
|
entities: list[EvoResetButtonBase] = [EvoResetSystemButton(coordinator, tcs)]
|
|
|
|
entities.extend(
|
|
[EvoResetZoneButton(coordinator, z) for z in tcs.zones if is_valid_zone(z)]
|
|
)
|
|
|
|
if tcs.hotwater:
|
|
entities.append(EvoResetDhwButton(coordinator, tcs.hotwater))
|
|
|
|
async_add_entities(entities)
|
|
|
|
for entity in entities:
|
|
await entity.update_attrs()
|
|
|
|
|
|
class EvoResetButtonBase(EvoEntity, ButtonEntity):
|
|
"""Button entity for system reset."""
|
|
|
|
_attr_entity_category = EntityCategory.CONFIG
|
|
|
|
_evo_state_attr_names = ()
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: EvoDataUpdateCoordinator,
|
|
evo_device: evo.ControlSystem | evo.HotWater | evo.Zone,
|
|
) -> None:
|
|
"""Initialize the system reset button."""
|
|
super().__init__(coordinator, evo_device)
|
|
|
|
# zones can be renamed, so set name in their property method
|
|
if isinstance(evo_device, evo.ControlSystem):
|
|
self._attr_name = f"Reset {evo_device.location.name}"
|
|
elif not isinstance(evo_device, evo.Zone):
|
|
self._attr_name = f"Reset {evo_device.name}"
|
|
|
|
self._attr_unique_id = f"{evo_device.id}_reset"
|
|
|
|
async def async_press(self) -> None:
|
|
"""Reset the Evohome entity to its base operating mode."""
|
|
await self.coordinator.call_client_api(self._evo_device.reset())
|
|
|
|
|
|
class EvoResetSystemButton(EvoResetButtonBase):
|
|
"""Button entity for system reset."""
|
|
|
|
_attr_translation_key = "reset_system_mode"
|
|
|
|
_evo_device: evo.ControlSystem
|
|
_evo_id_attr = "system_id"
|
|
|
|
|
|
class EvoResetDhwButton(EvoResetButtonBase):
|
|
"""Button entity for DHW override reset."""
|
|
|
|
_attr_translation_key = "clear_dhw_override"
|
|
|
|
_evo_device: evo.HotWater
|
|
_evo_id_attr = "dhw_id"
|
|
|
|
|
|
class EvoResetZoneButton(EvoResetButtonBase):
|
|
"""Button entity for zone override reset."""
|
|
|
|
_attr_translation_key = "clear_zone_override"
|
|
|
|
_evo_device: evo.Zone
|
|
_evo_id_attr = "zone_id"
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: EvoDataUpdateCoordinator,
|
|
evo_device: evo.Zone,
|
|
) -> None:
|
|
"""Initialize the zone reset button."""
|
|
super().__init__(coordinator, evo_device)
|
|
|
|
if evo_device.id == evo_device.tcs.id:
|
|
# this system does not have a distinct ID for the zone
|
|
self._attr_unique_id = f"{evo_device.id}z_reset"
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
"""Return the name of the evohome entity."""
|
|
return f"Reset {self._evo_device.name}"
|