Merge pull request #13 from manueliglesiasgarcia/master

Add page and page_size to find query
opus-42 · Apr 20, 2022 · 26a583a · 26a583a
2 parents e5f1e9e + 0327038
commit 26a583a
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/supersetapiclient/base.py b/supersetapiclient/base.py
@@ -258,26 +258,27 @@ def get(self, id: int):
 
         return object
 
-    def find(self, **kwargs):
+    def find(self, page_size: int = 100, page: int = 0, **kwargs):
         """Find and get objects from api."""
         url = self.base_url
 
         # Get response
-        if kwargs != {}:
-            query = {
+        query = {
+                "page_size": page_size,
+                "page": page,
                 "filters": [
                     {
                         "col": k,
                         "opr": "eq",
                         "value": v
                     } for k, v in kwargs.items()
                 ]
-            }
-            params = {
-                "q": json.dumps(query)
-            }
-        else:
-            params = {}
+        }
+
+        params = {
+            "q": json.dumps(query)
+        }
+
         response = self.client.get(
             url,
             params=params