bump

custom_components/kittybediq/button.py

@@ -0,0 +1,72 @@
+"""Support for SleepIQ buttons."""
+
+from collections.abc import Callable
+from dataclasses import dataclass
+from typing import Any, override
+
+from asyncsleepiq import SleepIQBed
+
+from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .coordinator import SleepIQConfigEntry
+from .entity import SleepIQEntity
+
+
+@dataclass(frozen=True, kw_only=True)
+class SleepIQButtonEntityDescription(ButtonEntityDescription):
+    """Class to describe a Button entity."""
+
+    press_action: Callable[[SleepIQBed], Any]
+
+
+ENTITY_DESCRIPTIONS = [
+    SleepIQButtonEntityDescription(
+        key="calibrate",
+        name="Calibrate",
+        press_action=lambda client: client.calibrate(),
+        icon="mdi:target",
+    ),
+    SleepIQButtonEntityDescription(
+        key="stop-pump",
+        name="Stop Pump",
+        press_action=lambda client: client.stop_pump(),
+        icon="mdi:stop",
+    ),
+]
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: SleepIQConfigEntry,
+    async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+    """Set up the sleep number buttons."""
+    data = entry.runtime_data
+
+    async_add_entities(
+        SleepNumberButton(bed, ed)
+        for bed in data.client.beds.values()
+        for ed in ENTITY_DESCRIPTIONS
+    )
+
+
+class SleepNumberButton(SleepIQEntity, ButtonEntity):
+    """Representation of an SleepIQ button."""
+
+    entity_description: SleepIQButtonEntityDescription
+
+    def __init__(
+        self, bed: SleepIQBed, entity_description: SleepIQButtonEntityDescription
+    ) -> None:
+        """Initialize the Button."""
+        super().__init__(bed)
+        self._attr_name = f"SleepNumber {bed.name} {entity_description.name}"
+        self._attr_unique_id = f"{bed.id}-{entity_description.key}"
+        self.entity_description = entity_description
+
+    @override
+    async def async_press(self) -> None:
+        """Press the button."""
+        await self.entity_description.press_action(self.bed)