tesla-fleet-api 1.0.10__py3-none-any.whl → 1.0.12__py3-none-any.whl

tesla_fleet_api/const.py

@@ -3,7 +3,7 @@
 from enum import Enum
 import logging
 
-VERSION = "1.0.10"
+VERSION = "1.0.12"
 LOGGER = logging.getLogger(__package__)
 SERVERS = {
     "na": "https://fleet-api.prd.na.vn.cloud.tesla.com",

tesla_fleet_api/tesla/bluetooth.py

@@ -1,10 +1,16 @@
 """Bluetooth only interface."""
 
+import asyncio
 import hashlib
 import re
+from bleak import BleakClient
+from bleak.backends.device import BLEDevice
+from bleak_retry_connector import establish_connection
 from google.protobuf.json_format import MessageToJson, MessageToDict
 
+from tesla_fleet_api.const import LOGGER
 from tesla_fleet_api.tesla.tesla import Tesla
+from tesla_fleet_api.tesla.vehicle.bluetooth import NAME_UUID
 from tesla_fleet_api.tesla.vehicle.vehicles import VehiclesBluetooth
 
 class TeslaBluetooth(Tesla):
@@ -27,6 +33,30 @@ class TeslaBluetooth(Tesla):
         """Get the name of a vehicle."""
         return "S" + hashlib.sha1(vin.encode('utf-8')).hexdigest()[:16] + "C"
 
+    async def query_display_name(self, device: BLEDevice, max_attempts=5) -> str | None:
+        """Queries the name of a bluetooth vehicle."""
+        client = await establish_connection(
+            BleakClient,
+            device,
+            device.name or "Unknown",
+            max_attempts=max_attempts
+        )
+        name: str | None = None
+        for i in range(max_attempts):
+            try:
+                # Standard GATT Device Name characteristic (0x2A00)
+                device_name = (await client.read_gatt_char(NAME_UUID)).decode('utf-8')
+                if device_name.startswith("🔑 "):
+                    name = device_name.replace("🔑 ","")
+                    break
+                await asyncio.sleep(1)
+                LOGGER.debug(f"Attempt {i+1} to query display name failed, {device_name}")
+            except Exception as e:
+                LOGGER.error(f"Failed to read device name: {e}")
+
+        await client.disconnect()
+        return name
+
 
 # Helpers
 

tesla_fleet_api/tesla/vehicle/bluetooth.py

@@ -2,7 +2,8 @@ from __future__ import annotations
 
 import hashlib
 import asyncio
-from typing import TYPE_CHECKING
+import struct
+from typing import TYPE_CHECKING, Callable, Any
 from google.protobuf.message import DecodeError
 from bleak_retry_connector import establish_connection, MAX_CONNECT_ATTEMPTS
 from bleak import BleakClient, BleakScanner
@@ -75,16 +76,79 @@ def prependLength(message: bytes) -> bytearray:
     """Prepend a 2-byte length to the payload."""
     return bytearray([len(message) >> 8, len(message) & 0xFF]) + message
 
+class ReassemblingBuffer:
+    """
+    Reassembles bytearray streams where the first two bytes indicate the length of the message.
+    Handles potential packet corruption by discarding *entire* packets and retrying.
+    Uses a callback to process parsed messages.
+    """
+
+    def __init__(self, callback: Callable[[RoutableMessage], None]):
+        """
+        Initializes the buffer.
+
+        Args:
+            message_type: The protobuf message type (e.g., RoutableMessage) to parse the assembled data.
+            callback: A function that will be called with each parsed message.
+        """
+        self.buffer = bytearray()
+        self.expected_length = None
+        self.packet_starts = []
+        self.callback = callback
+
+    def receive_data(self, data: bytearray):
+        """
+        Receives a chunk of bytearray data and attempts to assemble a complete message.
+
+        Args:
+            data: The received bytearray data.
+        """
+        self.packet_starts.append(len(self.buffer))
+        self.buffer.extend(data)
+
+        while True:
+            if self.expected_length is None and len(self.buffer) >= 2:
+                self.expected_length = struct.unpack(">H", self.buffer[:2])[0] + 2
+
+            LOGGER.info(f"Buffer length: {len(self.buffer)}, Packet starts: {self.packet_starts}, Expected length: {self.expected_length}")
+
+            if self.expected_length is not None and self.expected_length > 1024:
+                LOGGER.warning(f"Expected length too large: {self.expected_length}")
+                self.discard_packet()
+
+            elif self.expected_length is not None and len(self.buffer) >= self.expected_length:
+                try:
+                    message = RoutableMessage()
+                    message.ParseFromString(bytes(self.buffer[2:self.expected_length]))
+                    self.buffer = self.buffer[self.expected_length:]
+                    self.packet_starts = [x - self.expected_length for x in self.packet_starts if x >= self.expected_length]
+                    self.expected_length = None
+                    self.callback(message)  # Call the callback with the parsed message
+
+                except DecodeError:
+                    self.discard_packet()
+            else:
+                return
+
+    def discard_packet(self):
+        self.packet_starts.pop(0)
+        if len(self.packet_starts) > 0:
+            self.buffer = self.buffer[self.packet_starts[0]:]
+            self.packet_starts = [x - self.packet_starts[0] for x in self.packet_starts]
+        else:
+            self.buffer = bytearray()
+            self.packet_starts = []
+        self.expected_length = None
+
 class VehicleBluetooth(Commands):
     """Class describing the Tesla Fleet API vehicle endpoints and commands for a specific vehicle with command signing."""
 
     ble_name: str
-    device: BLEDevice
-    client: BleakClient
+    device: BLEDevice | None = None
+    client: BleakClient | None = None
     _queues: dict[Domain, asyncio.Queue]
     _ekey: ec.EllipticCurvePublicKey
-    _recv: bytearray = bytearray()
-    _recv_len: int = 0
+    _buffer: ReassemblingBuffer
     _auth_method = "aes"
 
     def __init__(
@@ -96,8 +160,9 @@ class VehicleBluetooth(Commands):
             Domain.DOMAIN_VEHICLE_SECURITY: asyncio.Queue(),
             Domain.DOMAIN_INFOTAINMENT: asyncio.Queue(),
         }
-        if device is not None:
-            self.device = device
+        self.device = device
+        self._connect_lock = asyncio.Lock()
+        self._buffer = ReassemblingBuffer(self._on_message)
 
     async def find_vehicle(self, name: str | None = None, address: str | None = None, scanner: BleakScanner | None = None) -> BLEDevice:
         """Find the Tesla BLE device."""
@@ -119,12 +184,12 @@ class VehicleBluetooth(Commands):
     def set_device(self, device: BLEDevice) -> None:
         self.device = device
 
-    def get_device(self) -> BLEDevice:
+    def get_device(self) -> BLEDevice | None:
         return self.device
 
     async def connect(self, max_attempts: int = MAX_CONNECT_ATTEMPTS) -> None:
         """Connect to the Tesla BLE device."""
-        if not hasattr(self, 'device'):
+        if not self.device:
             raise ValueError(f"BLEDevice {self.ble_name} has not been found or set")
         self.client = await establish_connection(
             BleakClient,
@@ -138,8 +203,17 @@ class VehicleBluetooth(Commands):
 
     async def disconnect(self) -> bool:
         """Disconnect from the Tesla BLE device."""
+        if not self.client:
+            return False
         return await self.client.disconnect()
 
+    async def connect_if_needed(self, max_attempts: int = MAX_CONNECT_ATTEMPTS) -> None:
+        """Connect to the Tesla BLE device if not already connected."""
+        async with self._connect_lock:
+            if not self.client or not self.client.is_connected:
+                LOGGER.info(f"Reconnecting to {self.ble_name}")
+                await self.connect(max_attempts=max_attempts)
+
     async def __aenter__(self) -> VehicleBluetooth:
         """Enter the async context."""
         await self.connect()
@@ -149,66 +223,44 @@ class VehicleBluetooth(Commands):
         """Exit the async context."""
         await self.disconnect()
 
-    async def _on_notify(self,sender: BleakGATTCharacteristic,data : bytearray) -> None:
+    def _on_notify(self,sender: BleakGATTCharacteristic, data: bytearray) -> None:
         """Receive data from the Tesla BLE device."""
-        if self._recv_len:
-            self._recv += data
-        else:
-            self._recv_len = int.from_bytes(data[:2], 'big')
-            if self._recv_len > 1024:
-                LOGGER.error("Parsed very large message length")
-                self._recv = bytearray()
-                self._recv_len = 0
-                return
-            self._recv = data[2:]
-        #while len(self._recv) > self._recv_len:
-        #
-        #    # Maybe this needs to trigger a reset
-        #    await self._on_message(bytes(self._recv[:self._recv_len]))
-        #    self._recv_len = int.from_bytes(self._recv[self._recv_len:self._recv_len+2], 'big')
-        #    self._recv = self._recv[self._recv_len+2:]
-        #    continue
-        if len(self._recv) >= self._recv_len:
-            if len(self._recv) > self._recv_len:
-                LOGGER.debug(f"Received more data than expected: {len(self._recv)} > {self._recv_len}")
-            try:
-                msg = RoutableMessage.FromString(bytes(self._recv[:self._recv_len]))
-                await self._on_message(msg)
-                self._recv = bytearray()
-                self._recv_len = 0
-            except DecodeError:
-                # Attempt parsing the whole payload
-                msg = RoutableMessage.FromString(bytes(self._recv))
-                LOGGER.warn(f"Parsed more data than length: {len(self._recv)} > {self._recv_len}")
-                await self._on_message(msg)
-                self._recv = bytearray()
-                self._recv_len = 0
-
-    async def _on_message(self, msg: RoutableMessage) -> None:
+        if sender.uuid != READ_UUID:
+            LOGGER.error(f"Unexpected sender: {sender}")
+            return
+        self._buffer.receive_data(data)
+
+    def _on_message(self, msg: RoutableMessage) -> None:
         """Receive messages from the Tesla BLE data."""
 
         if(msg.to_destination.routing_address != self._from_destination):
             # Ignore ephemeral key broadcasts
             return
 
-        LOGGER.debug(f"Received response: {msg}")
-        await self._queues[msg.from_destination.domain].put(msg)
+        LOGGER.info(f"Received response: {msg}")
+        self._queues[msg.from_destination.domain].put_nowait(msg)
 
-    async def _send(self, msg: RoutableMessage, requires: str, timeout: int = 2) -> RoutableMessage:
+    async def _send(self, msg: RoutableMessage, requires: str, timeout: int = 5) -> RoutableMessage:
         """Serialize a message and send to the vehicle and wait for a response."""
+
         domain = msg.to_destination.domain
         async with self._sessions[domain].lock:
-            LOGGER.debug(f"Sending message {msg}")
+            LOGGER.info(f"Sending message {msg}")
 
             payload = prependLength(msg.SerializeToString())
 
             # Empty the queue before sending the message
             while not self._queues[domain].empty():
-                await self._queues[domain].get()
+                msg = await self._queues[domain].get()
+                LOGGER.warning(f"Discarded message {msg}")
+
+            await self.connect_if_needed()
+            assert self.client is not None
             await self.client.write_gatt_char(WRITE_UUID, payload, True)
 
             # Process the response
             async with asyncio.timeout(timeout):
+                LOGGER.info(f"Waiting for response with {requires}")
                 while True:
                     resp = await self._queues[domain].get()
                     LOGGER.debug(f"Received message {resp}")
@@ -217,12 +269,16 @@ class VehicleBluetooth(Commands):
 
                     if resp.HasField(requires):
                         return resp
+                    else:
+                        LOGGER.warning(f"Ignoring message since it does not contain the required field {requires}, {resp.HasField(requires)}")
 
     async def query_display_name(self, max_attempts=5) -> str | None:
         """Read the device name via GATT characteristic if available"""
         for i in range(max_attempts):
             try:
                 # Standard GATT Device Name characteristic (0x2A00)
+                await self.connect_if_needed()
+                assert self.client
                 device_name = (await self.client.read_gatt_char(NAME_UUID)).decode('utf-8')
                 if device_name.startswith("🔑 "):
                     return device_name.replace("🔑 ","")
@@ -235,6 +291,8 @@ class VehicleBluetooth(Commands):
         """Read the device appearance via GATT characteristic if available"""
         try:
             # Standard GATT Appearance characteristic (0x2A01)
+            await self.connect_if_needed()
+            assert self.client
             return await self.client.read_gatt_char(APPEARANCE_UUID)
         except Exception as e:
             LOGGER.error(f"Failed to read device appearance: {e}")
@@ -244,6 +302,8 @@ class VehicleBluetooth(Commands):
         """Read the device version via GATT characteristic if available"""
         try:
             # Custom GATT Version characteristic (0x2A02)
+            await self.connect_if_needed()
+            assert self.client
             device_version = await self.client.read_gatt_char(VERSION_UUID)
             # Convert the bytes to an integer
             if device_version and len(device_version) > 0:
@@ -252,6 +312,11 @@ class VehicleBluetooth(Commands):
             LOGGER.error(f"Failed to read device version: {e}")
         return None
 
+    async def _command(self, domain: Domain, command: bytes, attempt: int = 0) -> dict[str, Any]:
+        """Serialize a message and send to the signed command endpoint."""
+        await self.connect_if_needed()
+        return await super()._command(domain, command, attempt)
+
     async def pair(self, role: Role = Role.ROLE_OWNER, form: KeyFormFactor = KeyFormFactor.KEY_FORM_FACTOR_CLOUD_KEY, timeout: int = 60):
         """Pair the key."""
 

tesla_fleet_api/tesla/vehicle/commands.py

@@ -269,7 +269,6 @@ class Commands(Vehicle):
         if msg.signedMessageStatus.signed_message_fault > 0:
             raise MESSAGE_FAULTS[msg.signedMessageStatus.signed_message_fault]
 
-    @abstractmethod
     async def _command(self, domain: Domain, command: bytes, attempt: int = 0) -> dict[str, Any]:
         """Serialize a message and send to the signed command endpoint."""
         session = self._sessions[domain]

tesla_fleet_api/tesla/vehicle/vehicles.py

@@ -73,27 +73,3 @@ class VehiclesBluetooth(dict[str, Vehicle]):
         vehicle = self.Bluetooth(self._parent, vin, key, device)
         self[vin] = vehicle
         return vehicle
-
-    async def query_display_name(self, device: BLEDevice, max_attempts=5) -> str | None:
-        """Queries the name of a bluetooth vehicle."""
-        client = await establish_connection(
-            BleakClient,
-            device,
-            device.name or "Unknown",
-            max_attempts=max_attempts
-        )
-        name: str | None = None
-        for i in range(max_attempts):
-            try:
-                # Standard GATT Device Name characteristic (0x2A00)
-                device_name = (await client.read_gatt_char(NAME_UUID)).decode('utf-8')
-                if device_name.startswith("🔑 "):
-                    name = device_name.replace("🔑 ","")
-                    break
-                await asyncio.sleep(1)
-                LOGGER.debug(f"Attempt {i+1} to query display name failed, {device_name}")
-            except Exception as e:
-                LOGGER.error(f"Failed to read device name: {e}")
-
-        await client.disconnect()
-        return name

{tesla_fleet_api-1.0.10.dist-info → tesla_fleet_api-1.0.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tesla_fleet_api
-Version: 1.0.10
+Version: 1.0.12
 Summary: Tesla Fleet API library for Python
 Home-page: https://github.com/Teslemetry/python-tesla-fleet-api
 Author: Brett Adams

{tesla_fleet_api-1.0.10.dist-info → tesla_fleet_api-1.0.12.dist-info}/RECORD

@@ -1,9 +1,9 @@
 tesla_fleet_api/__init__.py,sha256=3DZMoZ-5srW-7SooAjqcRubQDuZPY8rMKH7eqIp4qtg,392
-tesla_fleet_api/const.py,sha256=iX1zonX_WX3NYpR_I9RAoVx3eM2iLHQjSD8C3QcBdTk,3749
+tesla_fleet_api/const.py,sha256=GyNepuHCbWnq7w_1TgJT5PjBpTNQhoCZdWCwIPKz4a8,3749
 tesla_fleet_api/exceptions.py,sha256=GvjJBR77xGt2g3vhiAxcNtysj-e1Me7F-9PwO9dyzOo,35430
 tesla_fleet_api/ratecalculator.py,sha256=4lz8yruUeouHXh_3ezsXX-CTpIegp1T1J4VuRV_qdHA,1791
 tesla_fleet_api/tesla/__init__.py,sha256=Cvpqu8OaOFmbuwu9KjgYrje8eVluDp2IU_zwdtXbmO0,282
-tesla_fleet_api/tesla/bluetooth.py,sha256=vxDPTM-Uq9IAVJjK54-p_llj26hssEEXAQI5-lVGx8Q,1137
+tesla_fleet_api/tesla/bluetooth.py,sha256=lyPRVf1YdcElrYBsKOMCaLwMPE9rO7Iw1a6nE7VUZ94,2369
 tesla_fleet_api/tesla/charging.py,sha256=D7I7cAf-3-95sIjyP6wpVqCq9Cppj6U-VPFQGpQQ8bs,1704
 tesla_fleet_api/tesla/energysite.py,sha256=vStffklBQfQNAO_1wrHLFu7BlBCTVVbLh7_IrAUL3wg,6131
 tesla_fleet_api/tesla/fleet.py,sha256=zfmagXF4TbbVOPWcngCSKebaGB1daXbw8mTJ9o8einY,5432
@@ -12,12 +12,12 @@ tesla_fleet_api/tesla/partner.py,sha256=e-l6sEP6-IupjFEQieSUjhhvRXF3aL4ebPNahcGF
 tesla_fleet_api/tesla/tesla.py,sha256=Gs8-L3OsEMhs1N_vdDx-E46bOHKGowXTBxmRiP8NKh4,2391
 tesla_fleet_api/tesla/user.py,sha256=w8rwiAOIFjuDus8M0RpZ0wucJtw8kYFKtJfYVk7Ekr0,1194
 tesla_fleet_api/tesla/vehicle/__init__.py,sha256=3A5_wTQHofRShof4pUNOtF78-7lUh62uz2jq2ecnmRY,381
-tesla_fleet_api/tesla/vehicle/bluetooth.py,sha256=CJgyE99Q4ezWjDjEaRD5hA6_-bHPpztXXdQcKQh9GZ0,18334
-tesla_fleet_api/tesla/vehicle/commands.py,sha256=_LXAtuAIkeojJuw8k1QjEabAQQH2MLD5YZyUH58QoSg,51288
+tesla_fleet_api/tesla/vehicle/bluetooth.py,sha256=rMxKOrc8T3Wz_sbuGYFot4ffCbOVP2jFSCd5tVuSWv0,21035
+tesla_fleet_api/tesla/vehicle/commands.py,sha256=L3apfGwry-IBuSSLbuoPPjW_q4new2xTr6Iy_kUa2pA,51268
 tesla_fleet_api/tesla/vehicle/fleet.py,sha256=K9BVZj6CChJSDSMFroa7Cz0KrsYWj32ILtQumarkLaU,32080
 tesla_fleet_api/tesla/vehicle/signed.py,sha256=RUzVnZIfykz3YZW2gaxd1iaN1i8LkLaEoiXrbqZn9kg,1339
 tesla_fleet_api/tesla/vehicle/vehicle.py,sha256=IBvRO6qGkUf4bi-pFongA9fIe9iScvz_wGXtzEJXilY,870
-tesla_fleet_api/tesla/vehicle/vehicles.py,sha256=jLE_T_fWEwFjuYU5_AlJ0KhEvWAPx2wybENESksJ2R8,3777
+tesla_fleet_api/tesla/vehicle/vehicles.py,sha256=wU0wXa57BD7InaUC1xoRhVc7oI5CemsocSryuUItyGI,2792
 tesla_fleet_api/tesla/vehicle/proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tesla_fleet_api/tesla/vehicle/proto/__init__.pyi,sha256=qFXWNIgl71wB260u-XPzaAwWAHL6krw21q-aXnBtop0,252
 tesla_fleet_api/tesla/vehicle/proto/car_server_pb2.py,sha256=v_eb4NDIkx_ZYPYW29_EFRky5vQ4b2q14gwuQSbouYw,29202
@@ -44,8 +44,8 @@ tesla_fleet_api/teslemetry/vehicles.py,sha256=9nybVg7VHKLa2woMG6fzMmQP6xJIE_jdAd
 tesla_fleet_api/tessie/__init__.py,sha256=9lhQJaB6X4PObUL9QdaaZYqs2BxiTidu3zmHcBESLVw,78
 tesla_fleet_api/tessie/tessie.py,sha256=uhg0oOIxpwDvlvdBhKHeF3AGR2PzmdBgzh2-_EkmSq0,2617
 tesla_fleet_api/tessie/vehicles.py,sha256=gfEatilI_ct-R4CM5xYhrlduqCR9IHlyc56WmJf7v7k,1149
-tesla_fleet_api-1.0.10.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-tesla_fleet_api-1.0.10.dist-info/METADATA,sha256=Dr_QRBUiyoZ4PUgBBipQkoTBbqxW7qtE-vJ9GxzY8io,4383
-tesla_fleet_api-1.0.10.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-tesla_fleet_api-1.0.10.dist-info/top_level.txt,sha256=jeNbog_1saXBFrGpom9WyPWmilxsyP3szL_G7JLWQfM,16
-tesla_fleet_api-1.0.10.dist-info/RECORD,,
+tesla_fleet_api-1.0.12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+tesla_fleet_api-1.0.12.dist-info/METADATA,sha256=Z3qVhTn4BJs_P54gNAn4W02VUgvFhXkRh7yTveJvPn4,4383
+tesla_fleet_api-1.0.12.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+tesla_fleet_api-1.0.12.dist-info/top_level.txt,sha256=jeNbog_1saXBFrGpom9WyPWmilxsyP3szL_G7JLWQfM,16
+tesla_fleet_api-1.0.12.dist-info/RECORD,,