Upstream/aioesphomeapi
1
0
Fork 0
mirror of https://github.com/esphome/aioesphomeapi.git synced 2025-02-17 01:51:23 +01:00
Code Issues Projects Releases Wiki Activity

Change to event

Browse Source
This commit is contained in:
Jesse Hills 2022-09-28 12:52:36 +13:00
parent 49db2b021d
commit d17509be5b
No known key found for this signature in database
GPG Key ID: BEAAE804EFD8E83A
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
aioesphomeapi/client.py
View File

@ -452,13 +452,14 @@ class APIClient:
) -> Callable[[], None]: ) -> Callable[[], None]:
self._check_authenticated() self._check_authenticated()
fut = asyncio.get_event_loop().create_future() event = asyncio.Event()
def on_msg(msg: message.Message) -> None: def on_msg(msg: message.Message) -> None:
if isinstance(msg, BluetoothDeviceConnectionResponse): if isinstance(msg, BluetoothDeviceConnectionResponse):
resp = BluetoothDeviceConnection.from_pb(msg) resp = BluetoothDeviceConnection.from_pb(msg)
if address == resp.address: if address == resp.address:
on_bluetooth_connection_state(resp.connected, resp.mtu, resp.error) on_bluetooth_connection_state(resp.connected, resp.mtu, resp.error)
event.set()
assert self._connection is not None assert self._connection is not None
await self._connection.send_message_callback_response( await self._connection.send_message_callback_response(
@ -471,7 +472,7 @@ class APIClient:
try: try:
async with async_timeout.timeout(timeout): async with async_timeout.timeout(timeout):
await fut await event.wait()
except asyncio.TimeoutError as err: except asyncio.TimeoutError as err:
raise TimeoutAPIError("Timeout waiting for connect response") from err raise TimeoutAPIError("Timeout waiting for connect response") from err