Graph: Add cursor-based pagination

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sw-utils"
-version = "v0.12.0"
+version = "v0.12.1"
 description = "StakeWise Python utils"
 authors = ["StakeWise Labs <info@stakewise.io>"]
 license = "GPL-3.0-or-later"

sw_utils/graph/client.py

@@ -42,30 +42,37 @@ async def fetch_pages(
         query: DocumentNode,
         params: dict | None = None,
         page_size: int | None = None,
+        cursor_pagination: bool = False,
     ) -> list[dict]:
         """
         Fetches all pages of the query. Returns concatenated result.
+        Supports both offset-based and cursor-based pagination.
         """
         if page_size is None:
             page_size = self.page_size
 
         params = params.copy() if params else {}
 
-        skip = 0  # page offset
+        skip = 0
+        cursor = ''
         all_items = []
-
         while True:
-            params.update({'first': page_size, 'skip': skip})
+            if cursor_pagination:
+                params.update({'first': page_size, 'lastID': cursor})
+            else:
+                params.update({'first': page_size, 'skip': skip})
             res = await self.run_query(query, params)
-
             entity = list(res.keys())[0]
             items = res[entity]
             all_items.extend(items)
 
             if len(items) < page_size:
                 break
 
-            skip += page_size
+            if cursor_pagination:
+                cursor = items[-1]['id']
+            else:
+                skip += page_size
 
         return all_items