initial commit

custom_components/openhasp/button.py

@@ -0,0 +1,62 @@
+"""Support for current page numbers."""
+import logging
+
+from homeassistant.components.button import (
+    ButtonDeviceClass,
+    ButtonEntity,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.const import CONF_NAME
+from homeassistant.helpers.entity import EntityCategory
+
+
+from .common import HASPEntity
+from .const import CONF_HWID, CONF_TOPIC
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant, entry: ConfigEntry, async_add_entities
+):
+    """Set up Plate Relays as switch based on a config entry."""
+
+    async_add_entities(
+        [
+            HASPRestartButton(
+                entry.data[CONF_NAME],
+                entry.data[CONF_HWID],
+                entry.data[CONF_TOPIC],
+            )
+        ]
+    )
+
+    return True
+
+
+class HASPRestartButton(HASPEntity, ButtonEntity):
+    """Representation of page number."""
+
+    _attr_entity_category = EntityCategory.CONFIG
+    _attr_device_class = ButtonDeviceClass.RESTART
+
+    def __init__(self, name, hwid, topic) -> None:
+        """Initialize the Restart Button."""
+        super().__init__(name, hwid, topic, "restart")
+        self._attr_name = f"{name} restart"
+        self._available = True
+
+    async def async_press(self) -> None:
+        """Handle the button press."""
+        await self.hass.components.mqtt.async_publish(
+            self.hass,
+            f"{self._topic}/command/restart",
+            "",
+            qos=0,
+            retain=False,
+        )
+
+    async def refresh(self):
+        """Sync local state back to plate."""
+        self.async_write_ha_state()