button.py

Go to the documentation of this file.

"""Support for Dremel 3D Printer buttons."""
 
from __future__ import annotations
 
from collections.abc import Callable
from dataclasses import dataclass
 
from dremel3dpy import Dremel3DPrinter
 
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .coordinator import DremelConfigEntry
from .entity import Dremel3DPrinterEntity
 
 
@dataclass(frozen=True, kw_only=True)
class Dremel3DPrinterButtonEntityDescription(ButtonEntityDescription):
    """Describes a Dremel 3D Printer button entity."""
 
    press_fn: Callable[[Dremel3DPrinter], None]
 
 
BUTTON_TYPES: tuple[Dremel3DPrinterButtonEntityDescription, ...] = (
    Dremel3DPrinterButtonEntityDescription(
        key="cancel_job",
        translation_key="cancel_job",
        press_fn=lambda api: api.stop_print(),
    ),
    Dremel3DPrinterButtonEntityDescription(
        key="pause_job",
        translation_key="pause_job",
        press_fn=lambda api: api.pause_print(),
    ),
    Dremel3DPrinterButtonEntityDescription(
        key="resume_job",
        translation_key="resume_job",
        press_fn=lambda api: api.resume_print(),
    ),
)
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: DremelConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up Dremel 3D Printer control buttons."""
    async_add_entities(
        Dremel3DPrinterButtonEntity(config_entry.runtime_data, description)
        for description in BUTTON_TYPES
    )
 
 
class Dremel3DPrinterButtonEntity(Dremel3DPrinterEntity, ButtonEntity):
    """Represent a Dremel 3D Printer button."""
 
    entity_description: Dremel3DPrinterButtonEntityDescription
 
    def press(self) -> None:
        """Handle the button press."""
        # api does not care about the current state
        try:
            self.entity_descriptionentity_description.press_fn(self._api_api)
        except RuntimeError as ex:
            raise HomeAssistantError(
                "An error occurred while submitting command"
            ) from ex