"""Demo platform that offers a fake button entity.""" from __future__ import annotations from homeassistant.components import persistent_notification from homeassistant.components.button import ButtonEntity from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import DOMAIN async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, ) -> None: """Set up the demo button platform.""" async_add_entities( [ DemoButton( unique_id="push", device_name="Push", icon="mdi:gesture-tap-button", ), ] ) class DemoButton(ButtonEntity): """Representation of a demo button entity.""" _attr_has_entity_name = True _attr_should_poll = False def __init__( self, unique_id: str, device_name: str, icon: str, ) -> None: """Initialize the Demo button entity.""" self._attr_unique_id = unique_id self._attr_icon = icon self._attr_device_info = { "identifiers": {(DOMAIN, unique_id)}, "name": device_name, } async def async_press(self) -> None: """Send out a persistent notification.""" persistent_notification.async_create( self.hass, "Button pressed", title="Button" )