Upstream/aioesphomeapi
1
0
Fork 0
mirror of https://github.com/esphome/aioesphomeapi.git synced 2024-12-23 16:57:47 +01:00
Code Issues Projects Releases Wiki Activity

Cleanup bluetooth connection failure code (#495)

Browse Source
This commit is contained in:
J. Nick Koston 2023-07-19 14:25:32 -05:00 committed by GitHub
parent 821735e3a3
commit 2d89b9e267
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 58 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
aioesphomeapi/client.py
View File

@ -588,7 +588,7 @@ class APIClient:
"""Handle a timeout."""
if fut.done():
return
fut.set_exception(asyncio.TimeoutError())
fut.set_exception(asyncio.TimeoutError)
def _on_bluetooth_device_connection_response(
self,
@ -653,47 +653,33 @@ class APIClient:
msg_types,
)
timeout_handle = self._loop.call_later(
timeout, self._handle_timeout, connect_future
)
try:
loop = self._loop
timeout_handle = loop.call_later(timeout, self._handle_timeout, connect_future)
timeout_expired = False
connect_ok = False
try:
await connect_future
connect_ok = True
except asyncio.TimeoutError as err:
timeout_expired = True
# Disconnect before raising the exception to ensure
# the slot is recovered before the timeout is raised
# to avoid race were we run out even though we have a slot.
addr = to_human_readable_address(address)
if debug:
_LOGGER.debug(
"%s: Connecting timed out, waiting for disconnect", addr
_LOGGER.debug("%s: Connecting timed out, waiting for disconnect", addr)
disconnect_timed_out = (
not await self._bluetooth_device_disconnect_guard_timeout(
address, disconnect_timeout
)
disconnect_timed_out = False
try:
await self.bluetooth_device_disconnect(
address, timeout=disconnect_timeout
)
except TimeoutAPIError:
disconnect_timed_out = True
if debug:
_LOGGER.debug(
"%s: Disconnect timed out: %s", addr, disconnect_timed_out
)
finally:
try:
unsub()
except (KeyError, ValueError):
_LOGGER.warning(
"%s: Bluetooth device connection timed out but already unsubscribed "
"(likely due to unexpected disconnect)",
addr,
)
raise TimeoutAPIError(
f"Timeout waiting for connect response while connecting to {addr} "
f"after {timeout}s, disconnect timed out: {disconnect_timed_out}, "
f" after {disconnect_timeout}s"
) from err
except asyncio.CancelledError:
finally:
if not connect_ok:
try:
unsub()
except (KeyError, ValueError):
@ -701,12 +687,30 @@ class APIClient:
"%s: Bluetooth device connection canceled but already unsubscribed",
addr,
)
raise
finally:
if not timeout_expired:
timeout_handle.cancel()
return unsub
async def _bluetooth_device_disconnect_guard_timeout(
self, address: int, timeout: float
) -> bool:
"""Disconnect from a Bluetooth device and guard against timeout.
Return true if the disconnect was successful, false if it timed out.
"""
try:
await self.bluetooth_device_disconnect(address, timeout=timeout)
except TimeoutAPIError:
if _LOGGER.isEnabledFor(logging.DEBUG):
_LOGGER.debug(
"%s: Disconnect timed out: %s",
to_human_readable_address(address),
timeout,
)
return False
return True
async def bluetooth_device_pair(
self, address: int, timeout: float = DEFAULT_BLE_TIMEOUT
) -> BluetoothDevicePairing:

5
aioesphomeapi/connection.py
View File

@ -599,7 +599,7 @@ class APIConnection:
"""Handle a timeout."""
if fut.done():
return
fut.set_exception(asyncio.TimeoutError())
fut.set_exception(asyncio.TimeoutError)
def _handle_complex_message(
self,
@ -658,13 +658,16 @@ class APIConnection:
# the message could fail to be removed if the
# the await is cancelled
timeout_handle = self._loop.call_later(timeout, self._handle_timeout, fut)
timeout_expired = False
try:
await fut
except asyncio.TimeoutError as err:
timeout_expired = True
raise TimeoutAPIError(
f"Timeout waiting for response for {type(send_msg)} after {timeout}s"
) from err
finally:
if not timeout_expired:
timeout_handle.cancel()
for msg_type in msg_types:
message_handlers[msg_type].discard(on_message)