button.py

Go to the documentation of this file.

"""BleBox button entities implementation."""
 
from __future__ import annotations
 
import blebox_uniapi.button
 
from homeassistant.components.button import ButtonEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from . import BleBoxConfigEntry
from .entity import BleBoxEntity
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: BleBoxConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up a BleBox button entry."""
    entities = [
        BleBoxButtonEntity(feature)
        for feature in config_entry.runtime_data.features.get("buttons", [])
    ]
    async_add_entities(entities, True)
 
 
class BleBoxButtonEntity(BleBoxEntity[blebox_uniapi.button.Button], ButtonEntity):
    """Representation of BleBox buttons."""
 
    def __init__(self, feature: blebox_uniapi.button.Button) -> None:
        """Initialize a BleBox button feature."""
        super().__init__(feature)
        self._attr_icon_attr_icon = self.get_iconget_icon()
 
    def get_icon(self) -> str | None:
        """Return icon for endpoint."""
        if "up" in self._feature.query_string:
            return "mdi:arrow-up-circle"
        if "down" in self._feature.query_string:
            return "mdi:arrow-down-circle"
        if "fav" in self._feature.query_string:
            return "mdi:heart-circle"
        if "open" in self._feature.query_string:
            return "mdi:arrow-up-circle"
        if "close" in self._feature.query_string:
            return "mdi:arrow-down-circle"
        return None
 
    async def async_press(self) -> None:
        """Handle the button press."""
        await self._feature.set()