Update api_server.py

This commit is contained in:
imonlinux
2025-11-11 19:56:29 -08:00
committed by GitHub
parent 676591a75d
commit a74bf2ca1d
+5 -1
View File
@@ -9,6 +9,8 @@ from typing import TYPE_CHECKING, List, Optional
# pylint: disable=no-name-in-module # pylint: disable=no-name-in-module
from aioesphomeapi._frame_helper.packets import make_plain_text_packets from aioesphomeapi._frame_helper.packets import make_plain_text_packets
from aioesphomeapi.api_pb2 import ( # type: ignore[attr-defined] from aioesphomeapi.api_pb2 import ( # type: ignore[attr-defined]
AuthenticationRequest,
AuthenticationResponse,
DisconnectRequest, DisconnectRequest,
DisconnectResponse, DisconnectResponse,
HelloRequest, HelloRequest,
@@ -55,7 +57,9 @@ class APIServer(asyncio.Protocol):
) )
return return
if isinstance(msg_inst, DisconnectRequest): if isinstance(msg_inst, AuthenticationRequest):
self.send_messages([AuthenticationResponse()])
elif isinstance(msg_inst, DisconnectRequest):
self.send_messages([DisconnectResponse()]) self.send_messages([DisconnectResponse()])
_LOGGER.debug("Disconnect requested") _LOGGER.debug("Disconnect requested")
if self._transport: if self._transport: