auracaster/bumble_mirror
3
0
Fork 1
mirror of https://github.com/google/bumble.git synced 2026-05-10 04:18:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

device: add auto_restart mechanism to .start_discovery (default to True)

Browse Source
This commit is contained in:
Abel Lucas
2022-10-31 18:31:38 +00:00
parent 78534b659a
commit 51ddb36c91
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
bumble/device.py
View File

@@ -662,6 +662,7 @@ class Device(CompositeEventEmitter):
self.powered_on = False self.powered_on = False
self.advertising = False self.advertising = False
self.advertising_type = None self.advertising_type = None
self.auto_restart_inquiry = True
self.auto_restart_advertising = False self.auto_restart_advertising = False
self.command_timeout = 10 # seconds self.command_timeout = 10 # seconds
self.gatt_server = gatt_server.Server(self) self.gatt_server = gatt_server.Server(self)
@@ -1055,7 +1056,7 @@ class Device(CompositeEventEmitter):
if advertisement := accumulator.update(report): if advertisement := accumulator.update(report):
self.emit('advertisement', advertisement) self.emit('advertisement', advertisement)
async def start_discovery(self): async def start_discovery(self, auto_restart=True):
await self.send_command(HCI_Write_Inquiry_Mode_Command( await self.send_command(HCI_Write_Inquiry_Mode_Command(
inquiry_mode=HCI_EXTENDED_INQUIRY_MODE inquiry_mode=HCI_EXTENDED_INQUIRY_MODE
), check_result=True) ), check_result=True)
@@ -1069,10 +1070,13 @@ class Device(CompositeEventEmitter):
self.discovering = False self.discovering = False
raise HCI_StatusError(response) raise HCI_StatusError(response)
self.auto_restart_inquiry = auto_restart
self.discovering = True self.discovering = True
async def stop_discovery(self): async def stop_discovery(self):
if self.discovering:
await self.send_command(HCI_Inquiry_Cancel_Command(), check_result=True) await self.send_command(HCI_Inquiry_Cancel_Command(), check_result=True)
self.auto_restart_inquiry = True
self.discovering = False self.discovering = False
@host_event_handler @host_event_handler
@@ -1795,9 +1799,13 @@ class Device(CompositeEventEmitter):
@host_event_handler @host_event_handler
@AsyncRunner.run_in_task() @AsyncRunner.run_in_task()
async def on_inquiry_complete(self): async def on_inquiry_complete(self):
if self.discovering: if self.auto_restart_inquiry:
# Inquire again # Inquire again
await self.start_discovery() await self.start_discovery(auto_restart=True)
else:
self.auto_restart_inquiry = True
self.discovering = False
self.emit('inquiry_complete')
@host_event_handler @host_event_handler
@with_connection_from_handle @with_connection_from_handle