Handle autoping and autoclose parameters

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

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