python

custom_components/yandex_station/lock.py

@@ -0,0 +1,34 @@
+import logging
+
+from homeassistant.components.lock import LockEntity
+
+from .core.entity import YandexEntity
+from .hass import hass_utils
+
+_LOGGER = logging.getLogger(__name__)
+
+INCLUDE_CAPABILITIES = ("devices.capabilities.lock",)
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    entities = []
+
+    for quasar, device, config in hass_utils.incluce_devices(hass, entry):
+        for instance in device["capabilities"]:
+            if instance["type"] not in INCLUDE_CAPABILITIES:
+                continue
+            entities.append(YandexLock(quasar, device, instance))
+
+    async_add_entities(entities)
+
+
+# noinspection PyAbstractClass
+class YandexLock(LockEntity, YandexEntity):
+    def internal_update(self, capabilities: dict, properties: dict):
+        if value := capabilities.get("lock"):
+            self._attr_is_locked = value == "closed"
+
+    async def async_unlock(self):
+        await self.device_action("lock", "open")
+
+    async def async_lock(self):
+        await self.device_action("lock", "closed")