feat(callback): add set_callback method

Author: zHElEARN

examples/polar_h10.py

@@ -37,20 +37,8 @@ async def main():
     # Inspect the device details
     inspect(device)
 
-    def heartrate_callback(data: HRData):
-        console.print(f"[bold green]Received Data:[/bold green] {data}")
-
-    def data_callback(data: Union[ECGData, ACCData]):
-        """
-        Callback function to handle incoming data from the Polar device.
-
-        Args:
-            data (Union[ECGData, ACCData]): The data received from the Polar device.
-        """
-        console.print(f"[bold green]Received Data:[/bold green] {data}")
-
     # Establish connection to the Polar device
-    async with PolarDevice(device, data_callback, heartrate_callback) as polar_device:
+    async with PolarDevice(device) as polar_device:
         # Query available features
         available_features = await polar_device.available_features()
         inspect(available_features)
@@ -79,6 +67,20 @@ def data_callback(data: Union[ECGData, ACCData]):
             ],
         )
 
+        def heartrate_callback(data: HRData):
+            console.print(f"[bold green]Received Data:[/bold green] {data}")
+
+        def data_callback(data: Union[ECGData, ACCData]):
+            """
+            Callback function to handle incoming data from the Polar device.
+
+            Args:
+                data (Union[ECGData, ACCData]): The data received from the Polar device.
+            """
+            console.print(f"[bold green]Received Data:[/bold green] {data}")
+
+        polar_device.set_callback(data_callback, heartrate_callback)
+
         # Start data streams for ECG and ACC
         await polar_device.start_stream(ecg_settings)
         await polar_device.start_stream(acc_settings)

polar_python/device.py

@@ -148,6 +148,16 @@ async def stop_heartrate_stream(self) -> None:
                 f"Failed to stop heart rate stream: {str(e)}"
             ) from e
 
+    def set_callback(
+        self,
+        data_callback: Callable[
+            [Union[constants.ECGData, constants.ACCData]], None
+        ] = None,
+        heartrate_callback: Callable[[constants.HRData], None] = None,
+    ) -> None:
+        self._data_callback = data_callback
+        self._heartrate_callback = heartrate_callback
+
     def _handle_pmd_control(
         self, sender: BleakGATTCharacteristic, data: bytearray
     ) -> None: