Update Pager to take and pass arguments to underlying callable (tableau#451)

t8y8 · web-flow · commit d87600b479ce · 2019-06-25T10:41:47.000-07:00
This enables us to use `get` methods that take optional parameters, like `usage` on get Views.
diff --git a/tableauserverclient/server/pager.py b/tableauserverclient/server/pager.py
@@ -1,3 +1,5 @@
+from functools import partial
+
 from . import RequestOptions
 from . import Sort
 
@@ -11,13 +13,15 @@ class Pager(object):
     Will loop over anything that returns (List[ModelItem], PaginationItem).
     """
 
-    def __init__(self, endpoint, request_opts=None):
+    def __init__(self, endpoint, request_opts=None, **kwargs):
 
         if hasattr(endpoint, 'get'):
             # The simpliest case is to take an Endpoint and call its get
-            self._endpoint = endpoint.get
+            endpoint = partial(endpoint.get, **kwargs)
+            self._endpoint = endpoint
         elif callable(endpoint):
             # but if they pass a callable then use that instead (used internally)
+            endpoint = partial(endpoint, **kwargs)
             self._endpoint = endpoint
         else:
             # Didn't get something we can page over
