thermopro: add set_datetime service for TP358/TP393

Author: stephan48

src/thermopro_ble/__init__.py

@@ -15,6 +15,7 @@
 )
 
 from .parser import ThermoProBluetoothDeviceData
+from .device import ThermoProDevice
 
 __version__ = "0.10.0"
 

src/thermopro_ble/device.py

@@ -0,0 +1,49 @@
+from __future__ import annotations
+
+from datetime import datetime
+from struct import pack
+from uuid import UUID
+from contextlib import AsyncExitStack
+
+from bleak import BleakClient
+from bleak.exc import BleakError
+from bleak.backends.device import BLEDevice
+
+class ThermoProDevice:
+    datetime_uuid = UUID("00010203-0405-0607-0809-0a0b0c0d2b11")
+
+
+    def __init__(self, ble : BLEDevice):
+        self.ble = ble
+        self.stack = AsyncExitStack()
+        self.client = None
+
+
+    async def connect(self, timeout=30):
+        if not self.client:
+            self.client = await self.stack.enter_async_context(BleakClient(self.ble, timeout=timeout))
+        return self.client
+
+
+    def pack_datetime(self, dt: datetime, ampm=False):
+        # taken from https://github.com/koenvervloesem/bluetooth-clocks/blob/main/src/bluetooth_clocks/devices/thermopro.py
+        return pack(
+            "BBBBBBBBBB",
+            0xA5,
+            dt.year % 2000,
+            dt.month,
+            dt.day,
+            dt.hour,
+            dt.minute,
+            dt.second,
+            dt.weekday() + 1,  # Monday-Sunday -> 0-6
+            int(not ampm), # 1 means 24 hour format / 0 12 hour format
+            0x5A,
+        )
+
+
+    async def set_datetime(self, dt: datetime):
+        # taken from https://github.com/fuatakgun/generic_bt/blob/main/custom_components/generic_bt/binary_sensor.py
+        # copy over https://github.com/fuatakgun/generic_bt/blob/2349b00f9202e891cd247f13b63954065f07a2d0/custom_components/generic_bt/generic_bt_api/device.py#L47
+        client = await self.connect()
+        await client.write_gatt_char(ThermoProDevice.datetime_uuid, self.pack_datetime(dt), True)