jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 06:07:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add CLOSED as an expected type (#83180)

Browse Source
This commit is contained in:
Joakim Sørensen 2022-12-03 16:19:16 +01:00 committed by GitHub
parent 6cef37641c
commit 0dfc0af08c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/websocket_api/http.py
View File

@ -214,7 +214,7 @@ class WebSocketHandler:
disconnect_warn = "Did not receive auth message within 10 seconds" disconnect_warn = "Did not receive auth message within 10 seconds"
raise Disconnect from err raise Disconnect from err
if msg.type in (WSMsgType.CLOSE, WSMsgType.CLOSING): if msg.type in (WSMsgType.CLOSE, WSMsgType.CLOSED, WSMsgType.CLOSING):
raise Disconnect raise Disconnect
if msg.type != WSMsgType.TEXT: if msg.type != WSMsgType.TEXT:
@ -238,7 +238,7 @@ class WebSocketHandler:
while not wsock.closed: while not wsock.closed:
msg = await wsock.receive() msg = await wsock.receive()
if msg.type in (WSMsgType.CLOSE, WSMsgType.CLOSING): if msg.type in (WSMsgType.CLOSE, WSMsgType.CLOSED, WSMsgType.CLOSING):
break break
if msg.type != WSMsgType.TEXT: if msg.type != WSMsgType.TEXT: