35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
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")
|