Re-implement iteration for paging.Paged.

Author: brettcannon

msrest/paging.py

@@ -44,21 +44,17 @@ class Paged(collections.Iterable):
     def __init__(self, command, classes, raw_headers=None):
         self.next_link = ""
         self.current_page = []
+        self._current_page_iter_index = 0
         self._derserializer = Deserializer(classes)
         self._get_next = command
         self._response = None
         self._raw_headers = raw_headers
 
     def __iter__(self):
-        """Iterate over response items in current page, automatically
-        retrieves next page.
-        """
-        for i in self.current_page:
-            yield i
-
-        while self.next_link is not None:
-            for i in self.next():
-                yield i
+        """Return 'self'."""
+        # Since iteration mutates this object, consider it an iterator in-and-of
+        # itself.
+        return self
 
     @classmethod
     def _get_subtype_map(cls):
@@ -85,10 +81,21 @@ def reset(self):
         self.next_link = ""
         self.current_page = []
 
-    def next(self):
-        """Get next page."""
-        if self.next_link is None:
+    def __next__(self):
+        """Iterate through responses."""
+        # Storing the list iterator might work out better, but there's no
+        # guarantee that some code won't replace the list entirely with a copy,
+        # invalidating an list iterator that might be saved between iterations.
+        if self._current_page_iter_index < len(self.current_page):
+            response = self.current_page[self._current_page_iter_index]
+            self._current_page_iter_index += 1
+            return response
+        elif self.next_link is None:
             raise GeneratorExit("End of paging")
-        self._response = self._get_next(self.next_link)
-        self._derserializer(self, self._response)
-        return self.current_page
+        else:
+            self._current_page_iter_index = 0
+            self._response = self._get_next(self.next_link)
+            self._derserializer(self, self._response)
+            return self.__next__()
+
+    next = __next__  # Python 2 compatibility.