wbapi-async 0.7.7__tar.gz → 0.7.8__tar.gz

{wbapi_async-0.7.7 → wbapi_async-0.7.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbapi-async
-Version: 0.7.7
+Version: 0.7.8
 Summary: Fast, lightweight async client for the Wildberries Seller API
 Project-URL: Documentation, https://serdukow.github.io/wbapi-async/
 Project-URL: Repository, https://github.com/serdukow/wbapi-async

{wbapi_async-0.7.7 → wbapi_async-0.7.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "wbapi-async"
-version = "0.7.7"
+version = "0.7.8"
 description = "Fast, lightweight async client for the Wildberries Seller API"
 
 authors = [

{wbapi_async-0.7.7 → wbapi_async-0.7.8}/src/wbapi/utils/paginate.py

@@ -19,6 +19,24 @@ class PaginationStrategy:
         raise NotImplementedError
 
 
+class RrdIdBodyCursorStrategy(PaginationStrategy):
+    def detect(self, raw: Any, page: list[Any], body: Any, page_size: int, path: str) -> bool:
+        return bool(body is not None and page and isinstance(page[-1], dict) and "rrdId" in page[-1])
+
+    async def paginate(
+        self, result: list[Any], page: list[Any], raw: Any, request: _Requester, page_size: int, body: Any
+    ) -> list[Any]:
+        from .._method import _extract_list
+
+        while len(page) >= page_size:
+            rrd_id = page[-1]["rrdId"]
+            page = _extract_list(await request(extra_body={"rrdId": rrd_id})) or []
+            if not page:
+                break
+            result.extend(page)
+        return result
+
+
 class RrdIdCursorStrategy(PaginationStrategy):
     def detect(self, raw: Any, page: list[Any], body: Any, page_size: int, path: str) -> bool:
         return bool(page and isinstance(page[-1], dict) and "rrd_id" in page[-1])
@@ -147,6 +165,7 @@ class CardsListStrategy(PaginationStrategy):
 
 PAGINATION_STRATEGIES: list[PaginationStrategy] = [
     CardsListStrategy(),
+    RrdIdBodyCursorStrategy(),
     RrdIdCursorStrategy(),
     LastChangeDateStrategy(),
     BodyCursorStrategy(),