fix: unexpected ws closing

Damego · Jul 22, 2023 · 66a5102 · 66a5102
1 parent 64a46c3
commit 66a5102
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/anilibria/api/gateway/client.py b/anilibria/api/gateway/client.py
@@ -72,7 +72,7 @@ async def connect(self):
     async def _receive_data(self) -> dict | WSMessage:
         response = await self._connection.receive()
 
-        if response.type is WSMsgType.CLOSING:
+        if response.type in {WSMsgType.CLOSING, WSMsgType.CLOSED}:
             self._closed = True
             return response