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

Small cleanups to connection code (#554)

Browse Source
This commit is contained in:
J. Nick Koston 2023-09-24 23:26:52 +02:00 committed by GitHub
parent 223eb23c05
commit aecade1e1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
aioesphomeapi/connection.py
View File

@ -408,7 +408,8 @@ class APIConnection:
if not self._is_socket_open: if not self._is_socket_open:
return return
now = self._loop.time() loop = self._loop
now = loop.time()
if self._send_pending_ping: if self._send_pending_ping:
self.send_message(PING_REQUEST_MESSAGE) self.send_message(PING_REQUEST_MESSAGE)
@ -416,10 +417,10 @@ class APIConnection:
# Do not reset the timer if it's already set # Do not reset the timer if it's already set
# since the only thing we want to reset the timer # since the only thing we want to reset the timer
# is if we receive a pong. # is if we receive a pong.
self._pong_timer = self._loop.call_at( self._pong_timer = loop.call_at(
now + self._keep_alive_timeout, self._async_pong_not_received now + self._keep_alive_timeout, self._async_pong_not_received
) )
else: elif self._debug_enabled():
# #
# We haven't reached the ping response (pong) timeout yet # We haven't reached the ping response (pong) timeout yet
# and we haven't seen a response to the last ping # and we haven't seen a response to the last ping
@ -548,8 +549,7 @@ class APIConnection:
f"Connection isn't established yet ({self._connection_state})" f"Connection isn't established yet ({self._connection_state})"
) )
message_type = PROTO_TO_MESSAGE_TYPE.get(type(msg)) if not (message_type := PROTO_TO_MESSAGE_TYPE.get(type(msg))):
if not message_type:
raise ValueError(f"Message type id not found for type {type(msg)}") raise ValueError(f"Message type id not found for type {type(msg)}")
if self._debug_enabled(): if self._debug_enabled():
@ -623,7 +623,7 @@ class APIConnection:
if do_stop is None or do_stop(resp): if do_stop is None or do_stop(resp):
fut.set_result(None) fut.set_result(None)
async def send_message_await_response_complex( async def send_message_await_response_complex( # pylint: disable=too-many-locals
self, self,
send_msg: message.Message, send_msg: message.Message,
do_append: Callable[[message.Message], bool] | None, do_append: Callable[[message.Message], bool] | None,
@ -645,8 +645,9 @@ class APIConnection:
# sending the message and registering the handler # sending the message and registering the handler
self.send_message(send_msg) self.send_message(send_msg)
loop = self._loop
# Unsafe to await between sending the message and registering the handler # Unsafe to await between sending the message and registering the handler
fut: asyncio.Future[None] = self._loop.create_future() fut: asyncio.Future[None] = loop.create_future()
responses: list[message.Message] = [] responses: list[message.Message] = []
on_message = partial( on_message = partial(
self._handle_complex_message, fut, responses, do_append, do_stop self._handle_complex_message, fut, responses, do_append, do_stop
@ -663,9 +664,7 @@ class APIConnection:
# We must not await without a finally or # We must not await without a finally or
# the message could fail to be removed if the # the message could fail to be removed if the
# the await is cancelled # the await is cancelled
timeout_handle = self._loop.call_at( timeout_handle = loop.call_at(loop.time() + timeout, self._handle_timeout, fut)
self._loop.time() + timeout, self._handle_timeout, fut
)
timeout_expired = False timeout_expired = False
try: try:
await fut await fut
@ -720,7 +719,7 @@ class APIConnection:
"""Factory to make a packet processor.""" """Factory to make a packet processor."""
message_type_to_proto = MESSAGE_TYPE_TO_PROTO message_type_to_proto = MESSAGE_TYPE_TO_PROTO
debug_enabled = self._debug_enabled debug_enabled = self._debug_enabled
message_handlers = self._message_handlers message_handlers_get = self._message_handlers.get
internal_message_types = INTERNAL_MESSAGE_TYPES internal_message_types = INTERNAL_MESSAGE_TYPES
def _process_packet(msg_type_proto: int, data: bytes) -> None: def _process_packet(msg_type_proto: int, data: bytes) -> None:
@ -774,8 +773,7 @@ class APIConnection:
# since we know the connection is still alive # since we know the connection is still alive
self._send_pending_ping = False self._send_pending_ping = False
handlers = message_handlers.get(msg_type) if handlers := message_handlers_get(msg_type):
if handlers:
for handler in handlers.copy(): for handler in handlers.copy():
handler(msg) handler(msg)