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

Emit on_disconnection event on connection close in BfxWebsocketClient.

Browse Source
This commit is contained in:
Davide Casale
2023-04-19 03:53:33 +02:00
parent 986aa525d7
commit 3441d2af2f
1 changed files with 19 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -56,7 +56,7 @@ class BfxWebsocketClient:
MAXIMUM_CONNECTIONS_AMOUNT = 20 MAXIMUM_CONNECTIONS_AMOUNT = 20
EVENTS = [ EVENTS = [
"open", "subscribed", "authenticated", "wss-error", "open", "subscribed", "authenticated", "disconnection", "wss-error",
*PublicChannelsHandler.EVENTS, *PublicChannelsHandler.EVENTS,
*AuthenticatedChannelsHandler.EVENTS *AuthenticatedChannelsHandler.EVENTS
] ]
@@ -68,10 +68,10 @@ class BfxWebsocketClient:
self.event_emitter = AsyncIOEventEmitter() self.event_emitter = AsyncIOEventEmitter()
self.inputs = BfxWebsocketInputs(handle_websocket_input=self.__handle_websocket_input)
self.handler = AuthenticatedChannelsHandler(event_emitter=self.event_emitter) self.handler = AuthenticatedChannelsHandler(event_emitter=self.event_emitter)
self.inputs = BfxWebsocketInputs(handle_websocket_input=self.__handle_websocket_input)
if log_filename is None: if log_filename is None:
self.logger = ColorLogger("BfxWebsocketClient", level=log_level) self.logger = ColorLogger("BfxWebsocketClient", level=log_level)
else: self.logger = FileLogger("BfxWebsocketClient", level=log_level, filename=log_filename) else: self.logger = FileLogger("BfxWebsocketClient", level=log_level, filename=log_filename)
@@ -178,7 +178,8 @@ class BfxWebsocketClient:
try: try:
await _connection() await _connection()
except (websockets.exceptions.ConnectionClosedError, socket.gaierror) as error: except (websockets.exceptions.ConnectionClosedError, socket.gaierror) as error:
if isinstance(error, websockets.exceptions.ConnectionClosedError) and error.code in (1006, 1012): if isinstance(error, websockets.exceptions.ConnectionClosedError):
if error.code in (1006, 1012):
if error.code == 1006: if error.code == 1006:
self.logger.error("Connection lost: no close frame received " \ self.logger.error("Connection lost: no close frame received " \
"or sent (1006). Attempting to reconnect...") "or sent (1006). Attempting to reconnect...")
@@ -202,6 +203,9 @@ class BfxWebsocketClient:
else: raise error else: raise error
if not reconnection.status: if not reconnection.status:
self.event_emitter.emit("disconnection",
self.websocket.close_code, self.websocket.close_reason)
break break
async def __authenticate(self, api_key, api_secret, filters=None): async def __authenticate(self, api_key, api_secret, filters=None):