Source code for treq.response
from __future__ import absolute_import, division, print_function
from twisted.python.components import proxyForInterface
from twisted.web.iweb import IResponse
from requests.cookies import cookiejar_from_dict
from treq.content import collect, content, json_content, text_content
[docs]class _Response(proxyForInterface(IResponse)):
"""
A wrapper for :class:`twisted.web.iweb.IResponse` which manages cookies and
adds a few convenience methods.
"""
def __init__(self, original, cookiejar):
self.original = original
self._cookiejar = cookiejar
[docs] def collect(self, collector):
"""
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):
"""
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):
"""
Collect the response body as JSON per :func:`treq.json_content()`.
:rtype: Deferred that fires with the decoded JSON when the entire body
has been read.
"""
return json_content(self.original)
[docs] def text(self, encoding='ISO-8859-1'):
"""
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):
"""
Get a list of all responses that (such as intermediate redirects),
that ultimately ended in the current response. The responses are
ordered chronologically.
:returns: A `list` of :class:`~treq.response._Response` objects
"""
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):
"""
Get a copy of this response's cookies.
:rtype: :class:`requests.cookies.RequestsCookieJar`
"""
jar = cookiejar_from_dict({})
if self._cookiejar is not None:
for cookie in self._cookiejar:
jar.set_cookie(cookie)
return jar