Class PretimedHTTPAdapter

dwave/cloud/utils/http.py:33–58

Sets a default timeout for all adapter (think session) requests. It is overridden with per-request timeout. But it can not be reset back to infinite wait (``None``). Usage: s = requests.Session() s.mount("http://", PretimedHTTPAdapter(timeout=5)) s.mount("https:

class PretimedHTTPAdapter(requests.adapters.HTTPAdapter):
    """Sets a default timeout for all adapter (think session) requests. It is
    overridden with per-request timeout. But it can not be reset back to
    infinite wait (``None``).

    Usage:

        s = requests.Session()
        s.mount("http://", PretimedHTTPAdapter(timeout=5))
        s.mount("https://", PretimedHTTPAdapter(timeout=5))

        s.get('http://httpbin.org/delay/6')                 # -> timeouts after 5sec
        s.get('http://httpbin.org/delay/6', timeout=10)     # -> completes after 6sec

    The alternative is to set ``timeout`` on each request manually/explicitly,
    subclass ``Session``, or monkeypatch ``Session.request()``.
    """

    def __init__(self, timeout=None, *args, **kwargs):
        self.timeout = timeout
        super().__init__(*args, **kwargs)

    def send(self, *args, **kwargs):
        # can't use setdefault because caller always sets timeout kwarg
        kwargs['timeout'] = self.timeout
        return super().send(*args, **kwargs)


class TimeoutingHTTPAdapter(PretimedHTTPAdapter):