004f40ceb8
* adapters: print statement removal Signed-off-by: Mateusz Susik <mateusz.susik@sigmoidal.io> * test fix - read timeout added Signed-off-by: Mateusz Susik <mateusz.susik@sigmoidal.io>
17 lines
562 B
Python
17 lines
562 B
Python
from requests.adapters import HTTPAdapter
|
|
|
|
|
|
class TimeoutAdapter(HTTPAdapter):
|
|
"""HTTP adapter that adds timeout to each query."""
|
|
def __init__(self, timeout=None, *args, **kwargs):
|
|
"""HTTP adapter that adds timeout to each query.
|
|
|
|
:param timeout: Timeout that will be added to each query
|
|
"""
|
|
self.timeout = timeout
|
|
super(TimeoutAdapter, self).__init__(*args, **kwargs)
|
|
|
|
def send(self, *args, **kwargs):
|
|
kwargs['timeout'] = self.timeout
|
|
return super(TimeoutAdapter, self).send(*args, **kwargs)
|