Update Pager to handle un-paged lists (tableau#322)

t8y8 · web-flow · commit 6094eb31f9a3 · 2018-07-19T14:54:26.000-07:00
Not all endpoints support pagination on the server-side. We don't really give a way to predict what will or won't work with `Pager`, so I've updated it to check for a `total_available` amount. If that is `None` the endpoint doesn't support paging, and we can just drain the existing list and return. If the endpoint gets updated to support paging, it'll just magically start working.
diff --git a/tableauserverclient/server/pager.py b/tableauserverclient/server/pager.py
@@ -36,6 +36,13 @@ def __iter__(self):
         # Fetch the first page
         current_item_list, last_pagination_item = self._endpoint(self._options)
 
+        if last_pagination_item.total_available is None:
+            # This endpoint does not support pagination, drain the list and return
+            while current_item_list:
+                yield current_item_list.pop(0)
+
+            return
+
         # Get the rest on demand as a generator
         while self._count < last_pagination_item.total_available:
             if len(current_item_list) == 0:
