From 2ac391373329b2d8c67d34fd7c056ff9db16a5f9 Mon Sep 17 00:00:00 2001 From: Cory Benfield Date: Sat, 26 Jan 2013 15:19:01 +0000 Subject: [PATCH] Make Response objects iterable. --- requests/models.py | 4 ++++ test_requests.py | 6 ++++++ 2 files changed, 10 insertions(+) diff --git a/requests/models.py b/requests/models.py index 41eea5a..4570ad6 100644 --- a/requests/models.py +++ b/requests/models.py @@ -486,6 +486,10 @@ class Response(object): """Returns true if :attr:`status_code` is 'OK'.""" return self.ok + def __iter__(self): + """Returns the current object.""" + return self.iter_content(128) + @property def ok(self): try: diff --git a/test_requests.py b/test_requests.py index 54c732d..b9cfde0 100644 --- a/test_requests.py +++ b/test_requests.py @@ -325,6 +325,12 @@ class RequestsTestCase(unittest.TestCase): r = requests.get(httpbin('get')) self.assertTrue(r.elapsed.total_seconds() > 0.0) + def test_response_is_iterable(self): + r = requests.Response() + io = StringIO.StringIO('abc') + r.raw = io + self.assertTrue(iter(r).next()) + io.close() if __name__ == '__main__': unittest.main() -- 2.34.1