Update Client.py (#82)

Add timeout to all RPC calls
alienworks · Aug 20, 2018 · 2d1f9b0 · 2d1f9b0
1 parent fd8769c
commit 2d1f9b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neo-collector/insert/Client.py b/neo-collector/insert/Client.py
@@ -370,7 +370,7 @@ def __init__(self, client, address):
 
     def setup(self):
 
-        response = requests.post(self.addr, json={'jsonrpc': '2.0', 'method': GET_BLOCK_COUNT, 'params': [], 'id': 1})
+        response = requests.post(self.addr, json={'jsonrpc': '2.0', 'method': GET_BLOCK_COUNT, 'params': [], 'id': 1}, timeout=TIMEOUT)
         self.update_endpoint_details(response)
 
         if response.status_code == 200:
@@ -421,4 +421,4 @@ def __ge__(self, other):
         return self._compare(other) >= 0
 
     def __str__(self):
-        return "[%s] %s %s %s" % (self.addr, self.status, self.height, self.elapsed)
+        return "[%s] %s %s %s" % (self.addr, self.status, self.height, self.elapsed)