wyzeapy 0.5.27__tar.gz → 0.5.28__tar.gz

{wyzeapy-0.5.27 → wyzeapy-0.5.28}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: wyzeapy
-Version: 0.5.27
+Version: 0.5.28
 Summary: A library for interacting with Wyze devices
 License: GPL-3.0-only
 Author: Katie Mulliken
@@ -10,6 +10,7 @@ Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: aiodns (>=3.2.0,<4.0.0)
 Requires-Dist: aiohttp (>=3.11.12,<4.0.0)
 Requires-Dist: pycryptodome (>=3.21.0,<4.0.0)

{wyzeapy-0.5.27 → wyzeapy-0.5.28}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wyzeapy"
-version = "0.5.27"
+version = "0.5.28"
 description = "A library for interacting with Wyze devices"
 authors = ["Katie Mulliken <katie@mulliken.net>"]
 license = "GPL-3.0-only"

{wyzeapy-0.5.27 → wyzeapy-0.5.28}/src/wyzeapy/services/base_service.py

@@ -626,6 +626,25 @@ class BaseService:
 
         return response_json
 
+    async def _get_lock_ble_token(self, device: Device) -> Dict[str, Optional[Any]]:
+        await self._auth_lib.refresh_if_should()
+
+        url_path = "/openapi/lock/v1/ble/token"
+
+        payload = {
+            "uuid": device.mac
+        }
+
+        payload = ford_create_payload(self._auth_lib.token.access_token, payload, url_path, "get")
+
+        url = f"https://yd-saas-toc.wyzecam.com{url_path}"
+
+        response_json = await self._auth_lib.get(url, params=payload)
+
+        check_for_errors_lock(self, response_json)
+
+        return response_json
+
     async def _get_device_info(self, device: Device) -> Dict[Any, Any]:
         await self._auth_lib.refresh_if_should()
 

{wyzeapy-0.5.27 → wyzeapy-0.5.28}/src/wyzeapy/services/lock_service.py

@@ -4,7 +4,9 @@
 #  the license with this file. If not, please write to:
 #  katie@mulliken.net to receive a copy
 from .base_service import BaseService
+from ..const import FORD_APP_SECRET
 from ..types import Device, DeviceTypes
+from ..utils import wyze_decrypt_cbc
 
 
 class Lock(Device):
@@ -13,12 +15,19 @@ class Lock(Device):
     unlocking = False
     door_open = False
     trash_mode = False
+    ble_id = None
+    ble_token = None
 
 
 class LockService(BaseService):
     async def update(self, lock: Lock):
         device_info = await self._get_lock_info(lock)
         lock.raw_dict = device_info["device"]
+        if lock.product_model == "YD_BT1":
+            ble_token_info = await self._get_lock_ble_token(lock)
+            lock.raw_dict["token"] = ble_token_info["token"]
+            lock.ble_id = ble_token_info["token"]["id"]
+            lock.ble_token = wyze_decrypt_cbc(FORD_APP_SECRET[:16], ble_token_info["token"]["token"])
 
         lock.available = lock.raw_dict.get("onoff_line") == 1
         lock.door_open = lock.raw_dict.get("door_open_status") == 1

{wyzeapy-0.5.27 → wyzeapy-0.5.28}/src/wyzeapy/utils.py

@@ -4,6 +4,7 @@
 #  the license with this file. If not, please write to:
 #  katie@mulliken.net to receive a copy
 import base64
+import binascii
 import hashlib
 from typing import Dict, Any, List, Optional
 
@@ -66,6 +67,20 @@ def wyze_decrypt(key, enc):
     return decrypt_txt
 
 
+def wyze_decrypt_cbc(key: str, enc_hex_str: str) -> str:
+    key_hash = hashlib.md5(key.encode("utf-8")).digest()
+    
+    iv = b"0123456789ABCDEF"
+    cipher = AES.new(key_hash, AES.MODE_CBC, iv)
+    
+    encrypted_bytes = binascii.unhexlify(enc_hex_str)
+    decrypted_bytes = cipher.decrypt(encrypted_bytes)
+    
+    # PKCS5Padding
+    padding_length = decrypted_bytes[-1]
+    return decrypted_bytes[:-padding_length].decode()
+
+
 def create_password(password: str) -> str:
     hex1 = hashlib.md5(password.encode()).hexdigest()
     hex2 = hashlib.md5(hex1.encode()).hexdigest()