diff --git a/ucoinpy/api/bma/__init__.py b/ucoinpy/api/bma/__init__.py index 3004086a3ea49e4a974922861398c84433ba71ba..50d0235b17f9e219673fef39ef14a0a9ed6a32bc 100644 --- a/ucoinpy/api/bma/__init__.py +++ b/ucoinpy/api/bma/__init__.py @@ -184,7 +184,7 @@ class API(object): response = await aiohttp.post(self.reverse_url("http", path), data=kwargs, headers=self.headers) return response - def connect_ws(self, path): + def connect_ws(self, path, autoping=True, autoclose=True): """ Connect to a websocket in order to use API parameters @@ -192,7 +192,7 @@ class API(object): :return: """ url = self.reverse_url("ws", path) - return aiohttp.ws_connect(url) + return aiohttp.ws_connect(url, autoping=autoping, autoclose=autoclose) from . import network, blockchain, tx, wot, node, ud, ws diff --git a/ucoinpy/api/bma/ws/__init__.py b/ucoinpy/api/bma/ws/__init__.py index 67f2c74716e27a649eb2ee48229034b7b569c4a0..aab2e9a82a3a5a345b4b9eb354670142f9a1896a 100644 --- a/ucoinpy/api/bma/ws/__init__.py +++ b/ucoinpy/api/bma/ws/__init__.py @@ -32,8 +32,8 @@ class Block(Websocket): """Connect to block websocket.""" schema = _Block.schema - def connect(self): - r = self.connect_ws('/block') + def connect(self, autoping=True, autoclose=True): + r = self.connect_ws('/block', autoping=autoping, autoclose=autoclose) return r @@ -64,6 +64,6 @@ class Peer(Websocket): "required": ["version", "currency", "pubkey", "endpoints", "signature"] } - def connect(self): - r = self.connect_ws('/peer') + def connect(self, autoping=True, autoclose=True): + r = self.connect_ws('/peer', autoping=autoping, autoclose=autoclose) return r