From 4bb62a942fa162ccafd803fc3cdafc2396af416e Mon Sep 17 00:00:00 2001 From: Davide Casale Date: Mon, 28 Nov 2022 16:58:09 +0100 Subject: [PATCH] Allow users to use .on and .once as non-decorators methods. --- bfxapi/websocket/BfxWebsocketClient.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/bfxapi/websocket/BfxWebsocketClient.py b/bfxapi/websocket/BfxWebsocketClient.py index 14c3bb0..6a67982 100644 --- a/bfxapi/websocket/BfxWebsocketClient.py +++ b/bfxapi/websocket/BfxWebsocketClient.py @@ -125,22 +125,26 @@ class BfxWebsocketClient(object): if all(bucket.websocket != None and bucket.websocket.open == True for bucket in self.buckets): self.event_emitter.emit("open") - def on(self, event): + def on(self, event, callback = None): 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 != None: + return self.event_emitter.on(event, callback) + def handler(function): self.event_emitter.on(event, function) - return handler - def once(self, event): + def once(self, event, callback = None): 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 != None: + return self.event_emitter.once(event, callback) + def handler(function): self.event_emitter.once(event, function) - return handler class _BfxWebsocketBucket(object):