aljaz/bitfinex-api-py
1
0
Fork 0
mirror of https://github.com/aljazceru/bitfinex-api-py.git synced 2025-12-19 06:44:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add better handling for info code 20051.

Browse Source
This commit is contained in:
Davide Casale
2023-02-17 17:02:09 +01:00
parent 16d0ee525e
commit d72fcf3981
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
bfxapi/websocket/client/bfx_websocket_client.py
View File

@@ -108,7 +108,9 @@ class BfxWebsocketClient(object):
f"Update the library to the latest version to continue (client version: {BfxWebsocketClient.VERSION}, " + f"Update the library to the latest version to continue (client version: {BfxWebsocketClient.VERSION}, " +
f"server version: {message['version']}).") f"server version: {message['version']}).")
elif isinstance(message, dict) and message["event"] == "info" and message["code"] == 20051: elif isinstance(message, dict) and message["event"] == "info" and message["code"] == 20051:
raise websockets.ConnectionClosedError(rcvd=None, sent=None) rcvd = websockets.frames.Close(code=1012, reason="Stop/Restart Websocket Server (please reconnect).")
raise websockets.ConnectionClosedError(rcvd=rcvd, sent=None)
elif isinstance(message, dict) and message["event"] == "auth": elif isinstance(message, dict) and message["event"] == "auth":
if message["status"] == "OK": if message["status"] == "OK":
self.event_emitter.emit("authenticated", message); self.authentication = True self.event_emitter.emit("authenticated", message); self.authentication = True
@@ -130,9 +132,9 @@ class BfxWebsocketClient(object):
def next(self): def next(self):
backoff_delay = self.peek() backoff_delay = self.peek()
__backoff_delay = self.__backoff_delay * self.__backoff_factor __backoff_delay = self.__backoff_delay * self.__backoff_factor
self.__backoff_delay = min(__backoff_delay, _Delay.BACKOFF_MAX) self.__backoff_delay = min(__backoff_delay, _Delay.BACKOFF_MAX)
return backoff_delay return backoff_delay
def peek(self): def peek(self):
@@ -146,9 +148,14 @@ class BfxWebsocketClient(object):
try: try:
await _connection() await _connection()
except (websockets.ConnectionClosedError, socket.gaierror) as error: except (websockets.ConnectionClosedError, socket.gaierror) as error:
if isinstance(error, websockets.ConnectionClosedError) and error.code == 1006: if isinstance(error, websockets.ConnectionClosedError) and (error.code == 1006 or error.code == 1012):
self.logger.error("Connection lost: no close frame received " if error.code == 1006:
+ "or sent (1006). Attempting to reconnect...") self.logger.error("Connection lost: no close frame received "
+ "or sent (1006). Attempting to reconnect...")
if error.code == 1012:
self.logger.info("WSS server is about to restart, reconnection "
+ "required (client received 20051). Attempt in progress...")
reconnection = Reconnection(status=True, attempts=1, timestamp=datetime.now()); reconnection = Reconnection(status=True, attempts=1, timestamp=datetime.now());