fix: add retries to polling functionality

src/oralb_ble/parser.py

@@ -12,8 +12,12 @@
 from dataclasses import dataclass
 from enum import Enum, auto
 
-from bleak import BLEDevice
-from bleak_retry_connector import BleakClientWithServiceCache, establish_connection
+from bleak import BleakError, BLEDevice
+from bleak_retry_connector import (
+    BleakClientWithServiceCache,
+    establish_connection,
+    retry_bluetooth_connection_error,
+)
 from bluetooth_data_tools import short_address
 from bluetooth_sensor_state_data import BluetoothData
 from home_assistant_bluetooth import BluetoothServiceInfo
@@ -329,20 +333,13 @@ def poll_needed(
             update_interval = BRUSHING_UPDATE_INTERVAL_SECONDS
         return last_poll > update_interval
 
-    async def async_poll(self, ble_device: BLEDevice) -> SensorUpdate:
-        """
-        Poll the device to retrieve any values we can't get from passive listening.
-        """
-        client = await establish_connection(
-            BleakClientWithServiceCache, ble_device, ble_device.address
-        )
-        try:
-            battery_char = client.services.get_characteristic(CHARACTERISTIC_BATTERY)
-            battery_payload = await client.read_gatt_char(battery_char)
-            pressure_char = client.services.get_characteristic(CHARACTERISTIC_PRESSURE)
-            pressure_payload = await client.read_gatt_char(pressure_char)
-        finally:
-            await client.disconnect()
+    @retry_bluetooth_connection_error()
+    async def _get_payload(self, client: BleakClientWithServiceCache) -> None:
+        """Get the payload from the brush using its gatt_characteristics."""
+        battery_char = client.services.get_characteristic(CHARACTERISTIC_BATTERY)
+        battery_payload = await client.read_gatt_char(battery_char)
+        pressure_char = client.services.get_characteristic(CHARACTERISTIC_PRESSURE)
+        pressure_payload = await client.read_gatt_char(pressure_char)
         tb_pressure = ACTIVE_CONNECTION_PRESSURE.get(
             pressure_payload[0], f"unknown pressure {pressure_payload[0]}"
         )
@@ -356,4 +353,21 @@ async def async_poll(self, ble_device: BLEDevice) -> SensorUpdate:
             SensorDeviceClass.BATTERY,
             "Battery",
         )
+        _LOGGER.debug("Successfully read active gatt characters")
+
+    async def async_poll(self, ble_device: BLEDevice) -> SensorUpdate:
+        """
+        Poll the device to retrieve any values we can't get from passive listening.
+        """
+        _LOGGER.debug("Polling Oral-B device: %s", ble_device.address)
+        client = await establish_connection(
+            BleakClientWithServiceCache, ble_device, ble_device.address
+        )
+        try:
+            await self._get_payload(client)
+        except BleakError as err:
+            _LOGGER.warning(f"Reading gatt characters failed with err: {err}")
+        finally:
+            await client.disconnect()
+            _LOGGER.debug("Disconnected from active bluetooth client")
         return self._finish_update()