From 82770faad71c8d2a34d427fe864686f8b213b3f0 Mon Sep 17 00:00:00 2001
From: SchumyHao <schumyhaojl@126.com>
Date: Tue, 22 May 2018 16:40:11 +0800
Subject: [PATCH] Add Xiaomi Aqara Lock support (#14419)

---
 homeassistant/components/lock/xiaomi_aqara.py | 92 +++++++++++++++++++
 homeassistant/components/xiaomi_aqara.py      |  3 +-
 2 files changed, 94 insertions(+), 1 deletion(-)
 create mode 100644 homeassistant/components/lock/xiaomi_aqara.py

diff --git a/homeassistant/components/lock/xiaomi_aqara.py b/homeassistant/components/lock/xiaomi_aqara.py
new file mode 100644
index 00000000000..9b084a2bc55
--- /dev/null
+++ b/homeassistant/components/lock/xiaomi_aqara.py
@@ -0,0 +1,92 @@
+"""
+Support for Xiaomi Aqara Lock.
+
+For more details about this platform, please refer to the documentation at
+https://home-assistant.io/components/lock.xiaomi_aqara/
+"""
+import logging
+from homeassistant.components.xiaomi_aqara import (PY_XIAOMI_GATEWAY,
+                                                   XiaomiDevice)
+from homeassistant.components.lock import LockDevice
+from homeassistant.const import (STATE_LOCKED, STATE_UNLOCKED)
+from homeassistant.helpers.event import async_call_later
+from homeassistant.core import callback
+
+_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
+
+
+async def async_setup_platform(hass, config, async_add_devices,
+                               discovery_info=None):
+    """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))
+    async_add_devices(devices)
+
+
+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
diff --git a/homeassistant/components/xiaomi_aqara.py b/homeassistant/components/xiaomi_aqara.py
index 27bd496a3f0..ae3a4e0be72 100644
--- a/homeassistant/components/xiaomi_aqara.py
+++ b/homeassistant/components/xiaomi_aqara.py
@@ -139,7 +139,8 @@ def setup(hass, config):
     xiaomi.listen()
     _LOGGER.debug("Gateways discovered. Listening for broadcasts")
 
-    for component in ['binary_sensor', 'sensor', 'switch', 'light', 'cover']:
+    for component in ['binary_sensor', 'sensor', 'switch', 'light', 'cover',
+                      'lock']:
         discovery.load_platform(hass, component, DOMAIN, {}, config)
 
     def stop_xiaomi(event):