Source code for treq.response
from typing import Any, Callable, List
from requests.cookies import cookiejar_from_dict
from http.cookiejar import CookieJar
from twisted.internet.defer import Deferred
from twisted.python import reflect
from twisted.python.components import proxyForInterface
from twisted.web.iweb import UNKNOWN_LENGTH, IResponse
from treq.content import collect, content, json_content, text_content
[docs]
class _Response(proxyForInterface(IResponse)): # type: ignore
"""
A wrapper for :class:`twisted.web.iweb.IResponse` which manages cookies and
adds a few convenience methods.
"""
original: IResponse
_cookiejar: CookieJar
def __init__(self, original: IResponse, cookiejar: CookieJar):
self.original = original
self._cookiejar = cookiejar
def __repr__(self) -> str:
"""
Generate a representation of the response which includes the HTTP
status code, Content-Type header, and body size, if available.
"""
if self.original.length == UNKNOWN_LENGTH:
size = "unknown size"
else:
size = "{:,d} bytes".format(self.original.length)
# Display non-ascii bits of the content-type header as backslash
# escapes.
content_type_bytes = b", ".join(
self.original.headers.getRawHeaders(b"content-type", ())
)
content_type = repr(content_type_bytes).lstrip("b")[1:-1]
return "<{} {} '{:.40s}' {}>".format(
reflect.qual(self.__class__),
self.original.code,
content_type,
size,
)
[docs]
def collect(self, collector: Callable[[bytes], None]) -> "Deferred[None]":
"""
Incrementally collect the body of the response, per
:func:`treq.collect()`.
:param collector: A single argument callable that will be called
with chunks of body data as it is received.
:returns: A `Deferred` that fires when the entire body has been
received.
"""
return collect(self.original, collector)
[docs]
def content(self) -> "Deferred[bytes]":
"""
Read the entire body all at once, per :func:`treq.content()`.
:returns: A `Deferred` that fires with a `bytes` object when the entire
body has been received.
"""
return content(self.original)
[docs]
def json(self, **kwargs: Any) -> "Deferred[Any]":
"""
Collect the response body as JSON per :func:`treq.json_content()`.
:param kwargs: Any keyword arguments accepted by :py:func:`json.loads`
:rtype: Deferred that fires with the decoded JSON when the entire body
has been read.
"""
return json_content(self.original, **kwargs)
[docs]
def text(self, encoding: str = "ISO-8859-1") -> "Deferred[str]":
"""
Read the entire body all at once as text, per
:func:`treq.text_content()`.
:rtype: A `Deferred` that fires with a unicode string when the entire
body has been received.
"""
return text_content(self.original, encoding)
[docs]
def history(self) -> "List[_Response]":
"""
Get a list of all responses that (such as intermediate redirects),
that ultimately ended in the current response. The responses are
ordered chronologically.
"""
response = self
history = []
while response.previousResponse is not None:
history.append(_Response(response.previousResponse, self._cookiejar))
response = response.previousResponse
history.reverse()
return history
[docs]
def cookies(self) -> CookieJar:
"""
Get a copy of this response's cookies.
"""
# NB: This actually returns a RequestsCookieJar, but we type it as a
# regular CookieJar because we want to ditch requests as a dependency.
# Full deprecation deprecation will require a subclass or wrapper that
# warns about the RequestCookieJar extensions.
jar: CookieJar = cookiejar_from_dict({})
for cookie in self._cookiejar:
jar.set_cookie(cookie)
return jar