2019-02-13 20:21:14 +00:00
|
|
|
"""Support for Xiaomi Aqara locks."""
|
2018-05-22 08:40:11 +00:00
|
|
|
import logging
|
2019-03-21 05:56:46 +00:00
|
|
|
|
2018-05-22 08:40:11 +00:00
|
|
|
from homeassistant.components.lock import LockDevice
|
2019-03-21 05:56:46 +00:00
|
|
|
from homeassistant.const import STATE_LOCKED, STATE_UNLOCKED
|
2018-05-22 08:40:11 +00:00
|
|
|
from homeassistant.core import callback
|
2019-03-21 05:56:46 +00:00
|
|
|
from homeassistant.helpers.event import async_call_later
|
|
|
|
|
|
|
|
from . import PY_XIAOMI_GATEWAY, XiaomiDevice
|
2018-05-22 08:40:11 +00:00
|
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
FINGER_KEY = 'fing_verified'
|
|
|
|
PASSWORD_KEY = 'psw_verified'
|
|
|
|
CARD_KEY = 'card_verified'
|
|
|
|
VERIFIED_WRONG_KEY = 'verified_wrong'
|
|
|
|
|
|
|
|
ATTR_VERIFIED_WRONG_TIMES = 'verified_wrong_times'
|
|
|
|
|
|
|
|
UNLOCK_MAINTAIN_TIME = 5
|
|
|
|
|
|
|
|
|
2019-02-13 20:21:14 +00:00
|
|
|
async def async_setup_platform(
|
|
|
|
hass, config, async_add_entities, discovery_info=None):
|
2018-05-22 08:40:11 +00:00
|
|
|
"""Perform the setup for Xiaomi devices."""
|
|
|
|
devices = []
|
|
|
|
|
|
|
|
for gateway in hass.data[PY_XIAOMI_GATEWAY].gateways.values():
|
|
|
|
for device in gateway.devices['lock']:
|
|
|
|
model = device['model']
|
|
|
|
if model == 'lock.aq1':
|
|
|
|
devices.append(XiaomiAqaraLock(device, 'Lock', gateway))
|
2018-08-24 14:37:30 +00:00
|
|
|
async_add_entities(devices)
|
2018-05-22 08:40:11 +00:00
|
|
|
|
|
|
|
|
|
|
|
class XiaomiAqaraLock(LockDevice, XiaomiDevice):
|
|
|
|
"""Representation of a XiaomiAqaraLock."""
|
|
|
|
|
|
|
|
def __init__(self, device, name, xiaomi_hub):
|
|
|
|
"""Initialize the XiaomiAqaraLock."""
|
|
|
|
self._changed_by = 0
|
|
|
|
self._verified_wrong_times = 0
|
|
|
|
|
|
|
|
super().__init__(device, name, xiaomi_hub)
|
|
|
|
|
|
|
|
@property
|
|
|
|
def is_locked(self) -> bool:
|
|
|
|
"""Return true if lock is locked."""
|
|
|
|
if self._state is not None:
|
|
|
|
return self._state == STATE_LOCKED
|
|
|
|
|
|
|
|
@property
|
|
|
|
def changed_by(self) -> int:
|
|
|
|
"""Last change triggered by."""
|
|
|
|
return self._changed_by
|
|
|
|
|
|
|
|
@property
|
|
|
|
def device_state_attributes(self) -> dict:
|
|
|
|
"""Return the state attributes."""
|
|
|
|
attributes = {
|
|
|
|
ATTR_VERIFIED_WRONG_TIMES: self._verified_wrong_times,
|
|
|
|
}
|
|
|
|
return attributes
|
|
|
|
|
|
|
|
@callback
|
|
|
|
def clear_unlock_state(self, _):
|
|
|
|
"""Clear unlock state automatically."""
|
|
|
|
self._state = STATE_LOCKED
|
|
|
|
self.async_schedule_update_ha_state()
|
|
|
|
|
|
|
|
def parse_data(self, data, raw_data):
|
|
|
|
"""Parse data sent by gateway."""
|
|
|
|
value = data.get(VERIFIED_WRONG_KEY)
|
|
|
|
if value is not None:
|
|
|
|
self._verified_wrong_times = int(value)
|
|
|
|
return True
|
|
|
|
|
|
|
|
for key in (FINGER_KEY, PASSWORD_KEY, CARD_KEY):
|
|
|
|
value = data.get(key)
|
|
|
|
if value is not None:
|
|
|
|
self._changed_by = int(value)
|
|
|
|
self._verified_wrong_times = 0
|
|
|
|
self._state = STATE_UNLOCKED
|
|
|
|
async_call_later(self.hass, UNLOCK_MAINTAIN_TIME,
|
|
|
|
self.clear_unlock_state)
|
|
|
|
return True
|
|
|
|
|
|
|
|
return False
|