CodingPirates/pyscrlink
1
0
Fork 0
mirror of https://github.com/kawasaki/pyscrlink.git synced 2025-09-07 02:00:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

BLEDbusSession: implement notification reader

Browse Source 
Signed-off-by: Shin'ichiro Kawasaki <kawasaki@juno.dti.ne.jp>
This commit is contained in:
Shin'ichiro Kawasaki
2023-09-05 21:08:15 +09:00
parent 94a9555c4c
commit fe3cd35ac0
1 changed files with 32 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
pyscrlink/scratch_link.py
View File

@@ -34,6 +34,7 @@ from sdbus_async.bluez.gatt_api import (
GattServiceInterfaceAsync, GattServiceInterfaceAsync,
) )
from asyncio import sleep from asyncio import sleep
from os import dup, fdopen
import threading import threading
import time import time
@@ -312,6 +313,8 @@ class BLEDBusSession(Session):
self.services = {} self.services = {}
self.chars = {} self.chars = {}
self.chars_cache = {} self.chars_cache = {}
self.notification_fps = {}
self.notification_params = {}
self._connect_to_adapters() self._connect_to_adapters()
async def _get_characteristics(self, service_path): async def _get_characteristics(self, service_path):
@@ -360,21 +363,38 @@ class BLEDBusSession(Session):
char = self.chars_cache.get(id) char = self.chars_cache.get(id)
if char: if char:
return char return char
await self._get_services() for i in range(5):
btuuid = BTUUID(id) await self._get_services()
for char in self.chars.values(): btuuid = BTUUID(id)
raw_uuid = await char.uuid for char in self.chars.values():
char_uuid = BTUUID(raw_uuid) raw_uuid = await char.uuid
if char_uuid == btuuid: char_uuid = BTUUID(raw_uuid)
self.chars_cache[id] = char if char_uuid == btuuid:
return char self.chars_cache[id] = char
logger.error(f"Can not get characteristic: {id}") return char
logger.error(f"Can not get characteristic: {id}. Retry.")
logger.error(f"Abandoned to get characteristic: {id}.")
return None return None
async def _start_notification(self, char): async def _start_notification(self, sid, cid, char):
logger.debug('startNotification') logger.debug('startNotification')
(fd, mtu) = await char.acquire_notify({}) (fd, mtu) = await char.acquire_notify({})
logger.debug(f'fd={fd}') self.fd = dup(fd)
fp = fdopen(self.fd, mode='rb', buffering=0, newline=None)
self.notification_fps[self.fd] = fp
self.notification_params[self.fd] = { 'serviceId': sid,
'characteristicId': cid,
'encoding': 'base64' }
self.loop.add_reader(self.fd, self._read_notification, self.fd)
logger.debug(f'added notification reader: fd={self.fd}')
def _read_notification(self, *args):
fd = args[0]
fp = self.notification_fps[fd]
data = fp.read()
params = self.notification_params[fd].copy()
params['message'] = base64.standard_b64encode(data).decode('ascii')
self.loop.create_task(self._send_notification('characteristicDidChange', params))
def handle_request(self, method, params): def handle_request(self, method, params):
logger.debug("handle request") logger.debug("handle request")
@@ -424,7 +444,7 @@ class BLEDBusSession(Session):
message = base64.standard_b64encode(value).decode('ascii') message = base64.standard_b64encode(value).decode('ascii')
res['result'] = { 'message': message, 'encode': 'base64' } res['result'] = { 'message': message, 'encode': 'base64' }
if params.get('startNotifications') == True: if params.get('startNotifications') == True:
await self._start_notification(c) await self._start_notification(service_id, chara_id, c)
logger.debug(res) logger.debug(res)
return res return res