Skip to content

Commit

Permalink
Add timeout argument to _query method
Browse files Browse the repository at this point in the history
Default value is None, so the current default behavior remains unchanged
  • Loading branch information
moppymopperson committed Dec 24, 2017
1 parent be73ff5 commit 11ac129
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions krakenex/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ def load_key(self, path):
self.secret = f.readline().strip()
return

def _query(self, urlpath, data, headers=None):
def _query(self, urlpath, data, headers=None, timeout=None):
""" Low-level query handling.
.. note::
Expand All @@ -105,6 +105,9 @@ def _query(self, urlpath, data, headers=None):
:type data: dict
:param headers: (optional) HTTPS headers
:type headers: dict
:param timeout: (optional) if not None, a `requests.HTTPError` will be
thrown after `timeout` seconds if a response has not been received
:type timeout: number
:returns: :py:meth:`requests.Response.json`-deserialised Python object
:raises: :py:exc:`requests.HTTPError`: if response status not successful
Expand All @@ -116,7 +119,8 @@ def _query(self, urlpath, data, headers=None):

url = self.uri + urlpath

self.response = self.session.post(url, data = data, headers = headers)
self.response = self.session.post(url, data = data, headers = headers,
timeout = timeout)

if self.response.status_code not in (200, 201, 202):
self.response.raise_for_status()
Expand Down

0 comments on commit 11ac129

Please sign in to comment.