allow for both the old , way and the new way

DefectDojo · Jan 16, 2025 · 4899e27 · 4899e27
1 parent 948a4e8
commit 4899e27
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dojo/api_v2/prefetch/mixins.py b/dojo/api_v2/prefetch/mixins.py
@@ -6,7 +6,9 @@
 
 class PrefetchListMixin(ListModelMixin):
     def list(self, request, *args, **kwargs):
-        prefetch_params = request.GET.getlist("prefetch")
+        prefetch_params = request.GET.get("prefetch", "")
+        prefetch_params = prefetch_params.split(",") if "," in prefetch_params else request.GET.getlist("prefetch")
+
         prefetcher = _Prefetcher()
 
         # Apply the same operations as the standard list method defined in the
@@ -30,7 +32,9 @@ def list(self, request, *args, **kwargs):
 
 class PrefetchRetrieveMixin(RetrieveModelMixin):
     def retrieve(self, request, *args, **kwargs):
-        prefetch_params = request.GET.getlist("prefetch")
+        prefetch_params = request.GET.get("prefetch", "")
+        prefetch_params = prefetch_params.split(",") if "," in prefetch_params else request.GET.getlist("prefetch")
+
         prefetcher = _Prefetcher()
 
         entry = self.get_object()