Fix custom items_attribute in pagination

Author: ziima

ninja/pagination.py

@@ -97,7 +97,7 @@ def paginate_queryset(
         offset = pagination.offset
         limit: int = min(pagination.limit, settings.PAGINATION_MAX_LIMIT)
         return {
-            "items": queryset[offset : offset + limit],
+            self.items_attribute: queryset[offset : offset + limit],
             "count": self._items_count(queryset),
         }  # noqa: E203
 
@@ -114,7 +114,7 @@ async def apaginate_queryset(
         else:
             items = queryset[offset : offset + limit]
         return {
-            "items": items,
+            self.items_attribute: items,
             "count": await self._aitems_count(queryset),
         }  # noqa: E203
 
@@ -149,7 +149,7 @@ def paginate_queryset(
         page_size = self._get_page_size(pagination.page_size)
         offset = (pagination.page - 1) * page_size
         return {
-            "items": queryset[offset : offset + page_size],
+            self.items_attribute: queryset[offset : offset + page_size],
             "count": self._items_count(queryset),
         }  # noqa: E203
 
@@ -168,7 +168,7 @@ async def apaginate_queryset(
             items = queryset[offset : offset + page_size]
 
         return {
-            "items": items,
+            self.items_attribute: items,
             "count": await self._aitems_count(queryset),
         }  # noqa: E203
 

tests/test_pagination.py

@@ -99,6 +99,26 @@ def paginate_queryset(self, items, pagination: Input, request, **params):
         }
 
 
+class CustomItemsLimitOffsetPagination(LimitOffsetPagination):
+    """Minimal LimitOffsetPagination with custom items_attribute."""
+
+    items_attribute = "results"
+
+    class Output(Schema):
+        results: List[int]
+        count: int
+
+
+class CustomItemsPageNumberPagination(PageNumberPagination):
+    """Minimal PageNumberPagination with custom items_attribute."""
+
+    items_attribute = "results"
+
+    class Output(Schema):
+        results: List[int]
+        count: int
+
+
 @api.get("/items_1", response=List[int])
 @paginate  # WITHOUT brackets (should use default pagination)
 def items_1(request, **kwargs):
@@ -160,6 +180,18 @@ def items_10(request):
     return ITEMS
 
 
+@api.get("/items_11", response=List[int])
+@paginate(CustomItemsLimitOffsetPagination)
+def items_11(request):
+    return list(range(100))
+
+
+@api.get("/items_12", response=List[int])
+@paginate(CustomItemsPageNumberPagination)
+def items_12(request):
+    return list(range(100))
+
+
 client = TestClient(api)
 
 
@@ -563,6 +595,16 @@ def items_11(request, **kwargs):
     }
 
 
+def test_case11():
+    response = client.get("/items_11").json()
+    assert response == {"results": list(range(100)), "count": 100}
+
+
+def test_case12():
+    response = client.get("/items_12").json()
+    assert response == {"results": list(range(100)), "count": 100}
+
+
 def test_config_error_None():
     with pytest.raises(ConfigError):