Workflow

Signed-off-by: RISHI GARG <134256793+Rishi-source@users.noreply.github.com>

Author: Rishi-source

vulnerabilities/pagination_mixin.py

@@ -3,14 +3,14 @@ class PaginatedListViewMixin:
     max_page_size = 100
 
     PAGE_SIZE_CHOICES = [
-        {'value': 20, 'label': '20 per page'},
-        {'value': 50, 'label': '50 per page'}, 
-        {'value': 100, 'label': '100 per page'}
+        {"value": 20, "label": "20 per page"},
+        {"value": 50, "label": "50 per page"},
+        {"value": 100, "label": "100 per page"},
     ]
 
     def get_paginate_by(self, queryset=None):
         try:
-            page_size = int(self.request.GET.get('page_size', self.paginate_by))
+            page_size = int(self.request.GET.get("page_size", self.paginate_by))
             if page_size <= 0:
                 return self.paginate_by
             return min(page_size, self.max_page_size)
@@ -19,21 +19,22 @@ def get_paginate_by(self, queryset=None):
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-        
+
         current_page_size = self.get_paginate_by()
-        total_count = context['paginator'].count
-
-        context.update({
-            'current_page_size': current_page_size,
-            'page_size_choices': self.PAGE_SIZE_CHOICES,
-            'total_count': total_count,
-            'page_range': self._get_page_range(
-                context['paginator'], 
-                context['page_obj'].number
-            ),
-            'search': self.request.GET.get('search', '')
-        })
-        
+        total_count = context["paginator"].count
+
+        context.update(
+            {
+                "current_page_size": current_page_size,
+                "page_size_choices": self.PAGE_SIZE_CHOICES,
+                "total_count": total_count,
+                "page_range": self._get_page_range(
+                    context["paginator"], context["page_obj"].number
+                ),
+                "search": self.request.GET.get("search", ""),
+            }
+        )
+
         return context
 
     def _get_page_range(self, paginator, current_page, window=2):
@@ -42,16 +43,16 @@ def _get_page_range(self, paginator, current_page, window=2):
 
         pages = [1]
         if current_page > 3:
-            pages.append('...')
+            pages.append("...")
 
         start_page = max(2, current_page - window)
         end_page = min(paginator.num_pages - 1, current_page + window)
         pages.extend(range(start_page, end_page + 1))
-        
+
         if current_page < paginator.num_pages - 2:
-            pages.append('...')
-            
+            pages.append("...")
+
         if paginator.num_pages not in pages:
             pages.append(paginator.num_pages)
-            
+
         return pages

vulnerabilities/views.py

@@ -9,7 +9,7 @@
 import logging
 from datetime import datetime
 
-from .pagination_mixin import PaginatedListViewMixin
+from .mixins import PaginatedListViewMixin
 
 from cvss.exceptions import CVSS2MalformedError
 from cvss.exceptions import CVSS3MalformedError
@@ -68,6 +68,7 @@ class PackageSearch(PaginatedListViewMixin, ListView):
     """
     View for searching and displaying packages with pagination.
     """
+
     model = models.Package
     template_name = "packages.html"
     ordering = ["type", "namespace", "name", "version"]
@@ -86,10 +87,12 @@ def get_queryset(self, query=None):
             .order_by("package_url")
         )
 
+
 class VulnerabilitySearch(PaginatedListViewMixin, ListView):
     """
     View for searching and displaying vulnerabilities with pagination.
     """
+
     model = models.Vulnerability
     template_name = "vulnerabilities.html"
     ordering = ["vulnerability_id"]
@@ -102,7 +105,7 @@ def get_context_data(self, **kwargs):
     def get_queryset(self, query=None):
         query = query or self.request.GET.get("search") or ""
         return self.model.objects.search(query=query).with_package_counts()
-    
+
 
 class PackageDetails(DetailView):
     model = models.Package