From 4103e614750517677cdb350c1bbdeca3e60161d0 Mon Sep 17 00:00:00 2001
From: inso <insomniak.fr@gmaiL.com>
Date: Wed, 16 Mar 2016 18:07:53 +0100
Subject: [PATCH] Handle autoping and autoclose parameters
---
ucoinpy/api/bma/__init__.py | 4 ++--
ucoinpy/api/bma/ws/__init__.py | 8 ++++----
2 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/ucoinpy/api/bma/__init__.py b/ucoinpy/api/bma/__init__.py
index 3004086a..50d0235b 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 67f2c747..aab2e9a8 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
--
GitLab