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)
|