button.py

Go to the documentation of this file.

"""Obihai button module."""
 
from __future__ import annotations
 
from homeassistant.components.button import (
    ButtonDeviceClass,
    ButtonEntity,
    ButtonEntityDescription,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import entity_platform
 
from .connectivity import ObihaiConnection
from .const import DOMAIN, OBIHAI
 
BUTTON_DESCRIPTION = ButtonEntityDescription(
    key="reboot",
    name=f"{OBIHAI} Reboot",
    device_class=ButtonDeviceClass.RESTART,
    entity_category=EntityCategory.CONFIG,
)
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    entry: ConfigEntry,
    async_add_entities: entity_platform.AddEntitiesCallback,
) -> None:
    """Set up the Obihai sensor entries."""
 
    requester: ObihaiConnection = hass.data[DOMAIN][entry.entry_id]
 
    buttons = [ObihaiButton(requester)]
    async_add_entities(buttons, update_before_add=True)
 
 
class ObihaiButton(ButtonEntity):
    """Obihai Reboot button."""
 
    entity_description = BUTTON_DESCRIPTION
 
    def __init__(self, requester: ObihaiConnection) -> None:
        """Initialize monitor sensor."""
        self.requesterrequester = requester
        self._pyobihai_pyobihai = requester.pyobihai
        self._attr_unique_id_attr_unique_id = f"{requester.serial}-reboot"
 
    def press(self) -> None:
        """Press button."""
 
        if not self._pyobihai_pyobihai.call_reboot():
            raise HomeAssistantError("Reboot failed!")