Fix bug regarding closing websocket connection with .close coroutine method.

This commit is contained in:
Davide Casale
2022-11-23 16:28:05 +01:00
parent 16bbe19634
commit 45a9565cf4

View File

@@ -68,6 +68,7 @@ class BfxWebsocketClient(object):
elif isinstance(message, list) and (chanId := message[0]) == 0 and message[1] != HEARTBEAT:
self.handler.handle(message[1], message[2])
except websockets.ConnectionClosedError: continue
finally: await self.websocket.wait_closed(); break
async def __authenticate(self, API_KEY, API_SECRET, filter=None):
data = { "event": "auth", "filter": filter, "apiKey": API_KEY }
@@ -170,6 +171,7 @@ class _BfxWebsocketBucket(object):
elif isinstance(message, list) and (chanId := message[0]) and message[1] != HEARTBEAT:
self.handler.handle(self.chanIds[chanId], *message[1:])
except websockets.ConnectionClosedError: continue
finally: await self.websocket.wait_closed(); break
@_require_websocket_connection
async def _subscribe(self, channel, subId=None, **kwargs):