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

Remove BfxWebsocketClient::once. Replace all occurrences with BfxWebsocketClient::on. Add BfxWebsocketClient::ONCE_EVENTS class variable.

Browse Source
This commit is contained in:
Davide Casale
2023-04-19 04:20:53 +02:00
parent 3441d2af2f
commit 3d9e7c7b25
4 changed files with 15 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -55,8 +55,10 @@ class BfxWebsocketClient:
MAXIMUM_CONNECTIONS_AMOUNT = 20
ONCE_EVENTS = [ "open", "authenticated", "disconnection" ]
EVENTS = [
"open", "subscribed", "authenticated", "disconnection", "wss-error",
*ONCE_EVENTS, "subscribed", "wss-error",
*PublicChannelsHandler.EVENTS,
*AuthenticatedChannelsHandler.EVENTS
]
@@ -110,9 +112,6 @@ class BfxWebsocketClient:
def _on_wss_timeout():
on_timeout_event.set()
raise ReconnectionTimeoutError("Connection has been offline for too long " \
f"without being able to reconnect (wss_timeout: {self.wss_timeout}s).")
async def _connection():
nonlocal reconnection, timer, tasks
@@ -173,7 +172,8 @@ class BfxWebsocketClient:
await asyncio.sleep(delay.next())
if on_timeout_event.is_set():
break
raise ReconnectionTimeoutError("Connection has been offline for too long " \
f"without being able to reconnect (wss_timeout: {self.wss_timeout}s).")
try:
await _connection()
@@ -259,30 +259,18 @@ class BfxWebsocketClient:
raise EventNotSupported(f"Event <{event}> is not supported. To get a list " \
"of available events print BfxWebsocketClient.EVENTS")
def _register_event(event, function):
if event in BfxWebsocketClient.ONCE_EVENTS:
self.event_emitter.once(event, function)
else: self.event_emitter.on(event, function)
if callback is not None:
for event in events:
self.event_emitter.on(event, callback)
_register_event(event, callback)
if callback is None:
def handler(function):
for event in events:
self.event_emitter.on(event, function)
return handler
def once(self, *events, callback = None):
for event in events:
if event not in BfxWebsocketClient.EVENTS:
raise EventNotSupported(f"Event <{event}> is not supported. To get a list " \
"of available events print BfxWebsocketClient.EVENTS")
if callback is not None:
for event in events:
self.event_emitter.once(event, callback)
if callback is None:
def handler(function):
for event in events:
self.event_emitter.once(event, function)
_register_event(event, function)
return handler