Fix pagination

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>

Author: TG1999

vulnerabilities/api.py

@@ -12,6 +12,7 @@
 from django.db.models import Prefetch
 from django_filters import rest_framework as filters
 from packageurl import PackageURL
+from rest_framework import pagination
 from rest_framework import serializers
 from rest_framework import viewsets
 from rest_framework.decorators import action
@@ -175,9 +176,9 @@ def filter_purl(self, queryset, name, value):
 
 
 class PackageViewSet(viewsets.ReadOnlyModelViewSet):
-    queryset = Package.objects.all()
+    queryset = Package.objects.all().order_by("id")
     serializer_class = PackageSerializer
-    paginate_by = 50
+    pagination.PageNumberPagination.page_size = 10
     filter_backends = (filters.DjangoFilterBackend,)
     filterset_class = PackageFilterSet
 
@@ -249,10 +250,10 @@ def get_queryset(self):
                 queryset=self.get_fixed_packages_qs(),
                 to_attr="filtered_fixed_packages",
             )
-        )
+        ).order_by("id")
 
     serializer_class = VulnerabilitySerializer
-    paginate_by = 50
+    pagination.PageNumberPagination.page_size = 10
     filter_backends = (filters.DjangoFilterBackend,)
     filterset_class = VulnerabilityFilterSet
 
@@ -266,11 +267,13 @@ def filter_cpe(self, queryset, name, value):
 
 
 class CPEViewSet(viewsets.ReadOnlyModelViewSet):
-    queryset = Vulnerability.objects.filter(
-        vulnerabilityreference__reference_id__startswith="cpe"
-    ).distinct()
+    queryset = (
+        Vulnerability.objects.filter(vulnerabilityreference__reference_id__startswith="cpe")
+        .distinct()
+        .order_by("id")
+    )
     serializer_class = VulnerabilitySerializer
-    paginate_by = 50
+    pagination.PageNumberPagination.page_size = 10
     filter_backends = (filters.DjangoFilterBackend,)
     filterset_class = CPEFilterSet
 
@@ -284,8 +287,8 @@ def filter_alias(self, queryset, name, value):
 
 
 class AliasViewSet(viewsets.ReadOnlyModelViewSet):
-    queryset = Vulnerability.objects.all()
+    queryset = Vulnerability.objects.all().order_by("id")
     serializer_class = VulnerabilitySerializer
-    paginate_by = 50
+    pagination.PageNumberPagination.page_size = 10
     filter_backends = (filters.DjangoFilterBackend,)
     filterset_class = AliasFilterSet

vulnerabilities/models.py

@@ -83,7 +83,7 @@ def alias(self):
         Returns packages that first received patch against this vulnerability
         in their particular version history.
         """
-        return self.aliases.all()
+        return self.aliases.all().order_by("-alias")
 
     def __str__(self):
         return self.vulnerability_id

vulnerablecode/settings.py

@@ -160,5 +160,5 @@
         "rest_framework.filters.SearchFilter",
     ),
     "DEFAULT_PAGINATION_CLASS": "rest_framework.pagination.PageNumberPagination",
-    "PAGE_SIZE": 100,
+    "PAGE_SIZE": 10,
 }