Merge pull request #1143 from Lukasa/master

Response as iterator
This commit is contained in:
Kenneth Reitz
2013-01-28 20:52:17 -08:00
2 changed files with 10 additions and 0 deletions
+4
View File
@@ -486,6 +486,10 @@ class Response(object):
"""Returns true if :attr:`status_code` is 'OK'."""
return self.ok
def __iter__(self):
"""Allows you to use a response as an iterator."""
return self.iter_content(128)
@property
def ok(self):
try:
+6
View File
@@ -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(next(iter(r)))
io.close()
if __name__ == '__main__':
unittest.main()