Update lock handling

yalesmartalarmclient/client.py

@@ -129,16 +129,9 @@ def get_all_devices(self, retry: int = 3) -> dict[str, Any]:
             raise
         return cast(dict[str, Any], devices["data"])
 
-    def get_locks(self, retry: int = 3) -> list[YaleLock]:
+    def get_locks(self) -> list[YaleLock]:
         """Return all locks."""
-        try:
-            return list(self.lock_api.locks())
-        except Exception:
-            _LOGGER.debug("Retry %d on path self.lock_api.locks()", 4 - retry)
-            if retry > 0:
-                time.sleep(5)
-                return self.get_locks(retry - 1)
-            raise
+        return self.lock_api.locks
 
     def get_cycle(self, retry: int = 3) -> dict[str, Any]:
         """Return full cycle."""

yalesmartalarmclient/lock.py

@@ -248,8 +248,9 @@ class YaleDoorManAPI:
     def __init__(self, auth: YaleAuth) -> None:
         """Initialize the module."""
         self.auth = auth
+        self.locks = list(self.get_locks())
 
-    def locks(self) -> Iterator[YaleLock]:
+    def get_locks(self) -> Iterator[YaleLock]:
         """Iterate through the locks we have available for this user.
 
         Yields: the locks
@@ -269,6 +270,28 @@ def locks(self) -> Iterator[YaleLock]:
                 lock = YaleLock(device, lock_api=self)
                 yield lock
 
+    def update_locks(self, devices: list[dict[str, Any]]) -> None:
+        """Update the locks we have available for this user.
+
+        Args:
+            devices: list of device data
+
+        Returns:
+            None
+
+        Example:
+            >>> from yalesmartalarmclient.client import YaleSmartAlarmClient
+            >>> client = YaleSmartAlarmClient(username="", password="")
+            >>> client.update_locks(devices)
+
+        """
+
+        for device in devices:
+            if device["type"] == YaleLock.DEVICE_TYPE:
+                lock = self.get(device["name"])
+                if lock:
+                    lock.update(device)
+
     def get(self, name: str) -> YaleLock | None:
         """Get a single lock with matching name.
 
@@ -287,7 +310,7 @@ def get(self, name: str) -> YaleLock | None:
             myfrontdoor [YaleLockState.UNLOCKED]
 
         """
-        for lock in self.locks():
+        for lock in self.locks:
             if lock == name:
                 return lock
         return None