diff --git a/bfxapi/websocket/client/bfx_websocket_client.py b/bfxapi/websocket/client/bfx_websocket_client.py index e0f90e2..c59f1b1 100644 --- a/bfxapi/websocket/client/bfx_websocket_client.py +++ b/bfxapi/websocket/client/bfx_websocket_client.py @@ -11,7 +11,7 @@ from pyee.asyncio import AsyncIOEventEmitter from .bfx_websocket_bucket import _HEARTBEAT, F, _require_websocket_connection, BfxWebSocketBucket from .bfx_websocket_inputs import BfxWebSocketInputs -from ..handlers import PublicChannelsHandler, AuthenticatedEventsHandler +from ..handlers import PublicChannelsHandler, AuthEventsHandler from ..exceptions import WebSocketAuthenticationRequired, InvalidAuthenticationCredentials, EventNotSupported, \ ZeroConnectionsError, ReconnectionTimeoutError, OutdatedClientVersion @@ -57,14 +57,14 @@ class BfxWebSocketClient: ONCE_EVENTS = [ "open", "authenticated", "disconnection", - *AuthenticatedEventsHandler.ONCE_EVENTS + *AuthEventsHandler.ONCE_EVENTS ] EVENTS = [ "subscribed", "wss-error", *ONCE_EVENTS, *PublicChannelsHandler.EVENTS, - *AuthenticatedEventsHandler.ON_EVENTS + *AuthEventsHandler.ON_EVENTS ] def __init__(self, host, credentials, *, wss_timeout = 60 * 15, log_filename = None, log_level = "INFO"): @@ -76,7 +76,7 @@ class BfxWebSocketClient: self.event_emitter = AsyncIOEventEmitter() - self.handler = AuthenticatedEventsHandler(event_emitter=self.event_emitter) + self.handler = AuthEventsHandler(event_emitter=self.event_emitter) self.inputs = BfxWebSocketInputs(handle_websocket_input=self.__handle_websocket_input) diff --git a/bfxapi/websocket/handlers/__init__.py b/bfxapi/websocket/handlers/__init__.py index 98dadbb..23f5aad 100644 --- a/bfxapi/websocket/handlers/__init__.py +++ b/bfxapi/websocket/handlers/__init__.py @@ -1,2 +1,2 @@ from .public_channels_handler import PublicChannelsHandler -from .authenticated_events_handler import AuthenticatedEventsHandler +from .auth_events_handler import AuthEventsHandler diff --git a/bfxapi/websocket/handlers/authenticated_events_handler.py b/bfxapi/websocket/handlers/auth_events_handler.py similarity index 93% rename from bfxapi/websocket/handlers/authenticated_events_handler.py rename to bfxapi/websocket/handlers/auth_events_handler.py index b3e1e12..701e395 100644 --- a/bfxapi/websocket/handlers/authenticated_events_handler.py +++ b/bfxapi/websocket/handlers/auth_events_handler.py @@ -2,7 +2,7 @@ from ...types import serializers from ...types.serializers import _Notification -class AuthenticatedEventsHandler: +class AuthEventsHandler: __once_abbreviations = { "os": "order_snapshot", "ps": "position_snapshot", "fos": "funding_offer_snapshot", "fcs": "funding_credit_snapshot", "fls": "funding_loan_snapshot", "ws": "wallet_snapshot" @@ -49,9 +49,9 @@ class AuthenticatedEventsHandler: if abbrevation == "n": return self.__notification(stream) - for abbrevations, serializer in AuthenticatedEventsHandler.__serializers.items(): + for abbrevations, serializer in AuthEventsHandler.__serializers.items(): if abbrevation in abbrevations: - event = AuthenticatedEventsHandler.__abbreviations[abbrevation] + event = AuthEventsHandler.__abbreviations[abbrevation] if all(isinstance(substream, list) for substream in stream): return self.event_emitter.emit(event, [ serializer.parse(*substream) for substream in stream ])