Add scene support

custom_components/plejd/button.py

@@ -0,0 +1,47 @@
+import logging
+
+from homeassistant.components.button import ButtonEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+DOMAIN = "plejd"
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    scenes = hass.data[DOMAIN]["scenes"].get(config_entry.entry_id, [])
+
+    entities = []
+    for s in scenes:
+        button = PlejdSceneButton(s, config_entry.entry_id)
+        entities.append(button)
+    async_add_entities(entities, False)
+
+class PlejdSceneButton(ButtonEntity):
+
+    def __init__(self, device, entry_id):
+        super().__init__()
+        self.device = device
+        self.entry_id = entry_id
+
+    @property
+    def device_info(self):
+        return {
+            "identifiers": {(DOMAIN, f"{self.entry_id}:{self.device.index}")},
+            "name": self.device.name,
+            "manufacturer": "Plejd",
+            #"connections": ???,
+        }
+
+    @property
+    def has_entity_name(self):
+        return True
+    
+    @property
+    def name(self):
+        return None
+
+    @property
+    def unique_id(self):
+        return f"{self.entry_id}:{self.device.index}"
+
+    async def async_press(self):
+        await self.device.activate()