mirror of
https://github.com/home-assistant/core.git
synced 2026-01-29 09:50:43 +01:00
77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
"""Button platform for HDFury Integration."""
|
|
|
|
from collections.abc import Awaitable, Callable
|
|
from dataclasses import dataclass
|
|
|
|
from hdfury import HDFuryAPI, HDFuryError
|
|
|
|
from homeassistant.components.button import (
|
|
ButtonDeviceClass,
|
|
ButtonEntity,
|
|
ButtonEntityDescription,
|
|
)
|
|
from homeassistant.const import EntityCategory
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from .const import DOMAIN
|
|
from .coordinator import HDFuryConfigEntry
|
|
from .entity import HDFuryEntity
|
|
|
|
PARALLEL_UPDATES = 1
|
|
|
|
|
|
@dataclass(kw_only=True, frozen=True)
|
|
class HDFuryButtonEntityDescription(ButtonEntityDescription):
|
|
"""Description for HDFury button entities."""
|
|
|
|
press_fn: Callable[[HDFuryAPI], Awaitable[None]]
|
|
|
|
|
|
BUTTONS: tuple[HDFuryButtonEntityDescription, ...] = (
|
|
HDFuryButtonEntityDescription(
|
|
key="reboot",
|
|
device_class=ButtonDeviceClass.RESTART,
|
|
entity_category=EntityCategory.CONFIG,
|
|
press_fn=lambda client: client.issue_reboot(),
|
|
),
|
|
HDFuryButtonEntityDescription(
|
|
key="issue_hotplug",
|
|
translation_key="issue_hotplug",
|
|
entity_category=EntityCategory.CONFIG,
|
|
press_fn=lambda client: client.issue_hotplug(),
|
|
),
|
|
)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: HDFuryConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up buttons using the platform schema."""
|
|
|
|
coordinator = entry.runtime_data
|
|
|
|
async_add_entities(
|
|
HDFuryButton(coordinator, description) for description in BUTTONS
|
|
)
|
|
|
|
|
|
class HDFuryButton(HDFuryEntity, ButtonEntity):
|
|
"""HDFury Button Class."""
|
|
|
|
entity_description: HDFuryButtonEntityDescription
|
|
|
|
async def async_press(self) -> None:
|
|
"""Handle Button Press."""
|
|
|
|
try:
|
|
await self.entity_description.press_fn(self.coordinator.client)
|
|
except HDFuryError as error:
|
|
raise HomeAssistantError(
|
|
translation_domain=DOMAIN,
|
|
translation_key="communication_error",
|
|
) from error
|