91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
"""Support for Yale Access Bluetooth locks."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from yalexs_ble import ConnectionInfo, LockInfo, LockState, LockStatus
|
|
|
|
from homeassistant.components.lock import LockEntity
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from . import YALEXSBLEConfigEntry
|
|
from .entity import YALEXSBLEEntity
|
|
from .models import YaleXSBLEData
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: YALEXSBLEConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up locks."""
|
|
async_add_entities(
|
|
[YaleXSBLELock(entry.runtime_data), YaleXSBLESecureModeLock(entry.runtime_data)]
|
|
)
|
|
|
|
|
|
class YaleXSBLEBaseLock(YALEXSBLEEntity, LockEntity):
|
|
"""A yale xs ble lock."""
|
|
|
|
_secure_mode: bool = False
|
|
|
|
@callback
|
|
def _async_update_state(
|
|
self, new_state: LockState, lock_info: LockInfo, connection_info: ConnectionInfo
|
|
) -> None:
|
|
"""Update the state."""
|
|
self._attr_is_locked = False
|
|
self._attr_is_locking = False
|
|
self._attr_is_unlocking = False
|
|
self._attr_is_jammed = False
|
|
lock_state = new_state.lock
|
|
if lock_state is LockStatus.LOCKED:
|
|
self._attr_is_locked = not self._secure_mode
|
|
elif lock_state is LockStatus.LOCKING:
|
|
self._attr_is_locking = True
|
|
elif lock_state is LockStatus.UNLOCKING:
|
|
self._attr_is_unlocking = True
|
|
elif lock_state is LockStatus.SECUREMODE:
|
|
self._attr_is_locked = True
|
|
elif lock_state in (
|
|
LockStatus.UNKNOWN_01,
|
|
LockStatus.UNKNOWN_06,
|
|
):
|
|
self._attr_is_jammed = True
|
|
elif lock_state is LockStatus.UNKNOWN:
|
|
self._attr_is_locked = None
|
|
super()._async_update_state(new_state, lock_info, connection_info)
|
|
|
|
async def async_unlock(self, **kwargs: Any) -> None:
|
|
"""Unlock the lock."""
|
|
await self._device.unlock()
|
|
|
|
|
|
class YaleXSBLELock(YaleXSBLEBaseLock, LockEntity):
|
|
"""A yale xs ble lock not in secure mode."""
|
|
|
|
_attr_name = None
|
|
|
|
async def async_lock(self, **kwargs: Any) -> None:
|
|
"""Lock the lock."""
|
|
await self._device.lock()
|
|
|
|
|
|
class YaleXSBLESecureModeLock(YaleXSBLEBaseLock):
|
|
"""A yale xs ble lock in secure mode."""
|
|
|
|
_attr_entity_registry_enabled_default = False
|
|
_attr_translation_key = "secure_mode"
|
|
_secure_mode = True
|
|
|
|
def __init__(self, data: YaleXSBLEData) -> None:
|
|
"""Initialize the entity."""
|
|
super().__init__(data)
|
|
self._attr_unique_id = f"{self._device.address}_secure_mode"
|
|
|
|
async def async_lock(self, **kwargs: Any) -> None:
|
|
"""Lock the lock."""
|
|
await self._device.securemode()
|