Merge pull request #3 from jozefKruszynski/parse-auto-pull-spring

Author: zweckj

aiotedee/lock.py

@@ -34,6 +34,7 @@ def __init__(
         is_charging: bool = False,
         state_change_result: int = 0,
         is_enabled_pullspring: bool = False,
+        is_enabled_auto_pullspring: bool = False,
         duration_pullspring: int = 0,
     ) -> None:
         """Initialize a new lock."""
@@ -47,6 +48,7 @@ def __init__(
         self._state_change_result = state_change_result
         self._duration_pullspring = duration_pullspring
         self._is_enabled_pullspring = is_enabled_pullspring
+        self._is_enabled_auto_pullspring = is_enabled_auto_pullspring
 
     @property
     def lock_name(self) -> str:
@@ -137,6 +139,15 @@ def is_enabled_pullspring(self) -> bool:
     def is_enabled_pullspring(self, value: bool):
         self._is_enabled_pullspring = value
 
+    @property
+    def is_enabled_auto_pullspring(self) -> bool:
+        """Return true if the lock is charging."""
+        return bool(self._is_enabled_auto_pullspring)
+
+    @is_enabled_auto_pullspring.setter
+    def is_enabled_auto_pullspring(self, value: bool):
+        self._is_enabled_auto_pullspring = value    
+
     @property
     def duration_pullspring(self) -> int:
         """Return the duration of the pullspring."""
@@ -158,5 +169,6 @@ def to_dict(self) -> dict[str, str | int | bool | None]:
             "is_charging": self._is_charging,
             "state_change_result": self._state_change_result,
             "is_enabled_pullspring": self._is_enabled_pullspring,
+            "is_enabled_auto_pullspring": self._is_enabled_auto_pullspring,
             "duration_pullspring": self._duration_pullspring,
         }

aiotedee/tedee_client.py

@@ -141,6 +141,7 @@ async def get_locks(self) -> None:
             ) = self.parse_lock_properties(lock_json)
             (
                 is_enabled_pullspring,
+                is_enabled_auto_pullspring,
                 duration_pullspring,
             ) = self.parse_pull_spring_settings(lock_json)
 
@@ -154,6 +155,7 @@ async def get_locks(self) -> None:
                 is_charging,
                 state_change_result,
                 is_enabled_pullspring,
+                is_enabled_auto_pullspring,
                 duration_pullspring,
             )
 
@@ -203,6 +205,7 @@ async def sync(self) -> None:
             if local_call_success:
                 (
                     lock.is_enabled_pullspring,
+                    lock.is_enabled_auto_pullspring,
                     lock.duration_pullspring,
                 ) = self.parse_pull_spring_settings(lock_json)
 
@@ -354,8 +357,9 @@ def parse_pull_spring_settings(self, settings: dict):
         """Parse the pull spring settings"""
         device_settings = settings.get("deviceSettings", {})
         pull_spring_enabled = bool(device_settings.get("pullSpringEnabled", False))
+        pull_spring_auto_enabled = bool(device_settings.get("autoPullSpringEnabled", False))
         pull_spring_duration = device_settings.get("pullSpringDuration", 5)
-        return pull_spring_enabled, pull_spring_duration
+        return pull_spring_enabled, pull_spring_auto_enabled, pull_spring_duration
 
     def _calculate_secure_local_token(self) -> str:
         """Calculate the secure token"""