From acbed0bc576d9f513cafea5fa6e5ed0fd860cb7c Mon Sep 17 00:00:00 2001 From: Davide Casale Date: Tue, 22 Nov 2022 18:58:36 +0100 Subject: [PATCH] Fix small bug from merge conflicts resolving. --- bfxapi/websocket/BfxWebsocketClient.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/bfxapi/websocket/BfxWebsocketClient.py b/bfxapi/websocket/BfxWebsocketClient.py index 5347b0d..07a4c11 100644 --- a/bfxapi/websocket/BfxWebsocketClient.py +++ b/bfxapi/websocket/BfxWebsocketClient.py @@ -8,7 +8,7 @@ from .handlers import Channels, PublicChannelsHandler, AuthenticatedChannelsHand from .exceptions import ConnectionNotOpen, TooManySubscriptions, WebsocketAuthenticationRequired, InvalidAuthenticationCredentials, EventNotSupported, OutdatedClientVersion -from ..utils.logger import CustomLogger +from ..utils.logger import Formatter, CustomLogger HEARTBEAT = "hb" @@ -30,7 +30,7 @@ class BfxWebsocketClient(object): *AuthenticatedChannelsHandler.EVENTS ] - def __init__(self, host, buckets=5, API_KEY=None, API_SECRET=None): + def __init__(self, host, buckets=5, log_level = "INFO", API_KEY=None, API_SECRET=None): self.host, self.websocket, self.event_emitter = host, None, AsyncIOEventEmitter() self.API_KEY, self.API_SECRET, self.authentication = API_KEY, API_SECRET, False @@ -39,6 +39,8 @@ class BfxWebsocketClient(object): self.buckets = [ _BfxWebsocketBucket(self.host, self.event_emitter, self.__bucket_open_signal) for _ in range(buckets) ] + self.logger = CustomLogger("BfxWebsocketClient", logLevel=log_level) + async def start(self): tasks = [ bucket._connect(index) for index, bucket in enumerate(self.buckets) ]