Add buttons support

custom_components/yandex_station/__init__.py

@@ -48,12 +48,12 @@
 _LOGGER = logging.getLogger(__name__)
 
 PLATFORMS = [
+    "button",
     "climate",
     "light",
     "humidifier",
     "media_player",
     "number",
-    "remote",
     "select",
     "switch",
     "vacuum",

custom_components/yandex_station/button.py

@@ -0,0 +1,26 @@
+from homeassistant.components.button import ButtonEntity
+
+from .core.entity import YandexCustomEntity
+from .hass import hass_utils
+
+INCLUDE_CAPABILITIES = ("devices.capabilities.custom.button",)
+
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    entities = []
+
+    for quasar, device, config in hass_utils.incluce_devices(hass, entry):
+        if instances := config.get("capabilities"):
+            for instance in device["capabilities"]:
+                if instance["type"] not in INCLUDE_CAPABILITIES:
+                    continue
+                if instance["parameters"].get("instance", "on") in instances:
+                    entities.append(YandexCustomButton(quasar, device, instance))
+
+    async_add_entities(entities)
+
+
+# noinspection PyAbstractClass
+class YandexCustomButton(ButtonEntity, YandexCustomEntity):
+    async def async_press(self) -> None:
+        await self.quasar.device_action(self.device["id"], self.instance, True)

custom_components/yandex_station/core/yandex_quasar.py

@@ -419,10 +419,15 @@ async def get_device(self, deviceid: str):
         return resp
 
     async def device_action(self, deviceid: str, instance: str, value):
-        action = {
-            "type": IOT_TYPES[instance],
-            "state": {"instance": instance, "value": value},
-        }
+        action = {"state": {"instance": instance, "value": value}}
+
+        if instance in IOT_TYPES:
+            action["type"] = IOT_TYPES[instance]
+        elif instance.isdecimal():
+            action["type"] = "devices.capabilities.custom.button"
+        else:
+            return
+
         r = await self.session.post(
             f"{URL_USER}/devices/{deviceid}/actions", json={"actions": [action]}
         )

custom_components/yandex_station/hass/hass_utils.py

@@ -25,6 +25,7 @@
     "devices.types.iron",
     "devices.types.openable",
     "devices.types.openable.curtain",
+    "devices.types.other",
     "devices.types.pet_drinking_fountain",
     "devices.types.pet_feeder",
     "devices.types.washing_machine",
@@ -61,7 +62,7 @@ def incluce_devices(
     for conf in includes:
         for device in quasar.devices:
             if isinstance(conf, str):
-                if conf == device["name"] or conf == device["id"]:
+                if conf == device["id"] or conf == device["name"]:
                     conf = build_include_config(device)
                     devices.append((quasar, device, conf))
                     break