pstruebi/bumble_fork
1
0
Fork 0
forked from auracaster/bumble_mirror
Code Pull Requests Activity

Pandora: Handle exception in WaitSecurity()

Browse Source
This commit is contained in:
Josh Wu
2023-10-07 21:05:56 +08:00
parent aec5543081
commit c8b2804446
1 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
bumble/pandora/security.py
View File

@@ -450,21 +450,18 @@ class SecurityService(SecurityServicer):
'security_request': pair, 'security_request': pair,
} }
# register event handlers with contextlib.closing(EventWatcher()) as watcher:
for event, listener in listeners.items(): # register event handlers
connection.on(event, listener) for event, listener in listeners.items():
watcher.on(connection, event, listener)
# security level already reached # security level already reached
if self.reached_security_level(connection, level): if self.reached_security_level(connection, level):
return WaitSecurityResponse(success=empty_pb2.Empty()) return WaitSecurityResponse(success=empty_pb2.Empty())
self.log.debug('Wait for security...') self.log.debug('Wait for security...')
kwargs = {} kwargs = {}
kwargs[await wait_for_security] = empty_pb2.Empty() kwargs[await wait_for_security] = empty_pb2.Empty()
# remove event handlers
for event, listener in listeners.items():
connection.remove_listener(event, listener) # type: ignore
# wait for `authenticate` to finish if any # wait for `authenticate` to finish if any
if authenticate_task is not None: if authenticate_task is not None: