first commit

2024-01-11 15:10:51 +03:00
commit eddf07c652
602 changed files with 67154 additions and 0 deletions
@@ -0,0 +1,130 @@
+import logging
+
+from homeassistant.components.vacuum import StateVacuumEntity, SUPPORT_START, \
+    SUPPORT_STOP, SUPPORT_FAN_SPEED, SUPPORT_PAUSE, STATE_CLEANING, \
+    SUPPORT_BATTERY
+from homeassistant.const import STATE_IDLE, STATE_PAUSED
+
+from . import DOMAIN, CONF_INCLUDE, DATA_CONFIG, YandexQuasar
+
+_LOGGER = logging.getLogger(__name__)
+
+DEVICES = ['devices.types.vacuum_cleaner']
+
+
+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 = [
+        YandexVacuum(quasar, device)
+        for device in quasar.devices
+        if device['name'] in include and device['type'] in DEVICES
+    ]
+    async_add_entities(devices)
+
+
+# noinspection PyAbstractClass
+class YandexVacuum(StateVacuumEntity):
+    _battery = None
+    _fan_speed = None
+    _fan_speed_list = None
+    _state = None
+    _supported = 0
+
+    def __init__(self, quasar: YandexQuasar, device: dict):
+        self.quasar = quasar
+        self.device = device
+
+    @property
+    def unique_id(self):
+        return self.device['id'].replace('-', '')
+
+    @property
+    def name(self):
+        return self.device['name']
+
+    @property
+    def should_poll(self):
+        return True
+
+    @property
+    def state(self):
+        return self._state
+
+    @property
+    def supported_features(self):
+        return self._supported
+
+    @property
+    def battery_level(self):
+        return self._battery
+
+    @property
+    def fan_speed(self):
+        return self._fan_speed
+
+    @property
+    def fan_speed_list(self):
+        return self._fan_speed_list
+
+    async def async_start(self):
+        await self.quasar.device_action(self.device['id'], on=True)
+
+    async def async_stop(self, **kwargs):
+        await self.quasar.device_action(self.device['id'], on=False)
+
+    async def async_pause(self):
+        await self.quasar.device_action(self.device['id'], pause=True)
+
+    async def async_set_fan_speed(self, fan_speed, **kwargs):
+        await self.quasar.device_action(self.device['id'],
+                                        work_speed=fan_speed)
+
+    async def async_added_to_hass(self):
+        data = await self.quasar.get_device(self.device['id'])
+
+        for capability in data['capabilities']:
+            # TODO: type devices.capabilities.on_off don't have property
+            #  parameters instance
+            instance = capability['parameters'].get('instance', 'on')
+            if instance == 'on':
+                self._supported |= SUPPORT_START | SUPPORT_STOP
+
+            elif instance == 'work_speed':
+                self._supported |= SUPPORT_FAN_SPEED
+                self._fan_speed_list = [
+                    p['value'] for p in capability['parameters']['modes']
+                ]
+
+            elif instance == 'pause':
+                self._supported |= SUPPORT_PAUSE
+
+        for property_ in data['properties']:
+            instance = property_['parameters']['instance']
+            if instance == 'battery_level':
+                self._supported |= SUPPORT_BATTERY
+
+        await self.async_update(data)
+
+    async def async_update(self, data: dict = None):
+        if not data:
+            data = await self.quasar.get_device(self.device['id'])
+
+        for capability in data['capabilities']:
+            if not capability['retrievable']:
+                continue
+
+            instance = capability['state']['instance']
+            if instance == 'on':
+                self._state = (STATE_CLEANING if capability['state']['value']
+                               else STATE_IDLE)
+            elif instance == 'work_speed':
+                self._fan_speed = capability['state']['value']
+            elif instance == 'pause':
+                if capability['state']['value']:
+                    self._state = STATE_PAUSED
+
+        for property_ in data['properties']:
+            instance = property_['parameters']['instance']
+            if instance == 'battery_level':
+                self._battery = property_['state']['value']