mirror of
https://github.com/home-assistant/core.git
synced 2026-07-01 11:46:40 +01:00
Rayman223
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
"""Home Assistant component for accessing the Wallbox Portal API button."""
|
|
|
|
from collections.abc import Awaitable, Callable
|
|
from dataclasses import dataclass
|
|
|
|
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from .const import (
|
|
CHARGER_DATA_KEY,
|
|
CHARGER_RESUME_SCHEDULE_KEY,
|
|
CHARGER_SERIAL_NUMBER_KEY,
|
|
)
|
|
from .coordinator import WallboxConfigEntry, WallboxCoordinator
|
|
from .entity import WallboxEntity
|
|
|
|
|
|
@dataclass(frozen=True, kw_only=True)
|
|
class WallboxButtonEntityDescription(ButtonEntityDescription):
|
|
"""Describes Wallbox button entity."""
|
|
|
|
press_fn: Callable[[WallboxCoordinator], Awaitable[None]]
|
|
|
|
|
|
BUTTON_TYPES: dict[str, WallboxButtonEntityDescription] = {
|
|
CHARGER_RESUME_SCHEDULE_KEY: WallboxButtonEntityDescription(
|
|
key=CHARGER_RESUME_SCHEDULE_KEY,
|
|
translation_key=CHARGER_RESUME_SCHEDULE_KEY,
|
|
press_fn=lambda coordinator: coordinator.async_resume_schedule(),
|
|
),
|
|
}
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: WallboxConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Create wallbox button entities in HASS."""
|
|
coordinator: WallboxCoordinator = entry.runtime_data
|
|
async_add_entities(
|
|
[WallboxButton(coordinator, BUTTON_TYPES[CHARGER_RESUME_SCHEDULE_KEY])]
|
|
)
|
|
|
|
|
|
# Coordinator is used to centralize the data updates
|
|
PARALLEL_UPDATES = 0
|
|
|
|
|
|
class WallboxButton(WallboxEntity, ButtonEntity):
|
|
"""Representation of the Wallbox portal."""
|
|
|
|
entity_description: WallboxButtonEntityDescription
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: WallboxCoordinator,
|
|
description: WallboxButtonEntityDescription,
|
|
) -> None:
|
|
"""Initialize a Wallbox button."""
|
|
super().__init__(coordinator)
|
|
self.entity_description = description
|
|
self._attr_unique_id = (
|
|
f"{description.key}-"
|
|
f"{coordinator.data[CHARGER_DATA_KEY][CHARGER_SERIAL_NUMBER_KEY]}"
|
|
)
|
|
|
|
async def async_press(self) -> None:
|
|
"""Resume schedule and EcoSmart mode after a manual stop."""
|
|
await self.entity_description.press_fn(self.coordinator)
|