80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
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)
|