Add option for max_pages=None.

johncadengo · johncadengo · commit dd692da05d06 · 2014-07-06T22:12:54.000-04:00
In that case the generator will continue until pagination stops.
diff --git a/instagram/bind.py b/instagram/bind.py
@@ -162,7 +162,7 @@ def _do_api_request(self, url, method="GET", body=None, headers=None):
         def _paginator_with_url(self, url, method="GET", body=None, headers=None):
             headers = headers or {}
             pages_read = 0
-            while url and pages_read < self.max_pages:
+            while url and (pages_read < self.max_pages or self.max_pages is None):
                 api_responses, url = self._do_api_request(url, method, body, headers)
                 pages_read += 1
                 yield api_responses, url
diff --git a/tests.py b/tests.py
@@ -126,6 +126,16 @@ def test_generator_user_feed(self):
         for page in generator:
             str(generator)
 
+    def test_generator_user_feed_all(self):
+        generator = self.api.user_media_feed(as_generator=True, max_pages=None)
+        for i in range(10):
+            page = generator.next()
+            str(generator)
+
+        generator = self.api.user_media_feed(as_generator=True, max_pages=0)
+        for page in generator:
+            assert False
+
     def test_user_liked_media(self):
         self.api.user_liked_media(count=10)
 
