Allow passing starting cursor in paginated helper (#222)

* Allow passing starting cursor in paginated helper Allow passing next_cursor to the paginated api to set the starting cursor. * Take a slightly different approach --------- Co-authored-by: ramnes <[email protected]>
ramnes · Dec 25, 2023 · c1faf0a · c1faf0a
1 parent 3e2a51f
commit c1faf0a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notion_client/helpers.py b/notion_client/helpers.py
@@ -30,7 +30,7 @@ def iterate_paginated_api(
     function: Callable[..., Any], **kwargs: Any
 ) -> Generator[List[Any], None, None]:
     """Return an iterator over the results of any paginated Notion API."""
-    next_cursor = None
+    next_cursor = kwargs.pop("start_cursor", None)
 
     while True:
         response = function(**kwargs, start_cursor=next_cursor)
@@ -53,7 +53,7 @@ async def async_iterate_paginated_api(
     function: Callable[..., Awaitable[Any]], **kwargs: Any
 ) -> AsyncGenerator[List[Any], None]:
     """Return an async iterator over the results of any paginated Notion API."""
-    next_cursor = None
+    next_cursor = kwargs.pop("start_cursor", None)
 
     while True:
         response = await function(**kwargs, start_cursor=next_cursor)