python

custom_components/ttlock/lock.py

@@ -0,0 +1,48 @@
+"""The actual lock part of the locks."""
+from __future__ import annotations
+
+from typing import Any
+
+from homeassistant.components.lock import LockEntity
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .coordinator import lock_coordinators
+from .entity import BaseLockEntity
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up all the locks for the config entry."""
+
+    async_add_entities(
+        [Lock(coordinator) for coordinator in lock_coordinators(hass, entry)]
+    )
+
+
+class Lock(BaseLockEntity, LockEntity):
+    """The entity object for a lock."""
+
+    def _update_from_coordinator(self) -> None:
+        """Fetch state from the device."""
+        self._attr_name = self.coordinator.data.name
+
+        self._attr_is_locked = self.coordinator.data.locked
+        self._attr_is_locking = (
+            self.coordinator.data.action_pending and not self.coordinator.data.locked
+        )
+        self._attr_is_unlocking = (
+            self.coordinator.data.action_pending and self.coordinator.data.locked
+        )
+
+    async def async_lock(self, **kwargs: Any) -> None:
+        """Try to lock the lock."""
+        await self.coordinator.lock()
+
+    async def async_unlock(self, **kwargs: Any) -> None:
+        """Try to unlock the lock."""
+        await self.coordinator.unlock()