first commit

hass/settings/custom_components/yandex_station/remote.py

@@ -0,0 +1,79 @@
+import asyncio
+import logging
+from typing import Iterable, Any
+
+from homeassistant.components.remote import RemoteEntity, ATTR_DELAY_SECS, \
+    ATTR_NUM_REPEATS
+
+from . import DOMAIN, DATA_CONFIG, CONF_INCLUDE, YandexQuasar
+
+_LOGGER = logging.getLogger(__name__)
+
+DEVICES = ['devices.types.other']
+
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    include = hass.data[DOMAIN][DATA_CONFIG][CONF_INCLUDE]
+    quasar = hass.data[DOMAIN][entry.unique_id]
+    devices = [
+        YandexOther(quasar, device)
+        for device in quasar.devices
+        if device['name'] in include and device['type'] in DEVICES
+    ]
+    async_add_entities(devices, True)
+
+
+# noinspection PyAbstractClass
+class YandexOther(RemoteEntity):
+    _name = None
+
+    def __init__(self, quasar: YandexQuasar, device: dict):
+        self.quasar = quasar
+        self.device = device
+
+        self.buttons = {}
+
+    async def async_added_to_hass(self):
+        self._name = self.device['name']
+
+        data = await self.quasar.get_device(self.device['id'])
+        for capability in data['capabilities']:
+            if capability['type'] != 'devices.capabilities.custom.button':
+                continue
+            name = capability['parameters']['name']
+            self.buttons[name] = capability['parameters']['instance']
+
+    @property
+    def unique_id(self):
+        return self.device['id'].replace('-', '')
+
+    @property
+    def name(self):
+        return self._name
+
+    @property
+    def should_poll(self) -> bool:
+        return False
+
+    @property
+    def is_on(self) -> bool:
+        return True
+
+    async def async_send_command(self, command: Iterable[str],
+                                 **kwargs: Any) -> None:
+        num_repeats = kwargs.get(ATTR_NUM_REPEATS)
+        if num_repeats:
+            command *= num_repeats
+
+        delay = kwargs.get(ATTR_DELAY_SECS, 0)
+
+        for i, cmd in enumerate(command):
+            if cmd not in self.buttons:
+                _LOGGER.error(f"Неизвестная команда {cmd}")
+                continue
+
+            if delay and i:
+                await asyncio.sleep(delay)
+
+            payload = {self.buttons[cmd]: True}
+            await self.quasar.device_action(self.device['id'], **payload)