From 22decc3f7f5b7f1974b9d0b87762eda2242b1ecf Mon Sep 17 00:00:00 2001
From: Cody Maffucci <46459665+Maffooch@users.noreply.github.com>
Date: Tue, 17 Sep 2024 19:34:55 -0500
Subject: [PATCH] Prefetch Serialization: Add a preference during mapping

---
 dojo/api_v2/prefetch/prefetcher.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/dojo/api_v2/prefetch/prefetcher.py b/dojo/api_v2/prefetch/prefetcher.py
index 79a4b0e731..3596b3f940 100644
--- a/dojo/api_v2/prefetch/prefetcher.py
+++ b/dojo/api_v2/prefetch/prefetcher.py
@@ -3,11 +3,17 @@
 
 from rest_framework.serializers import ModelSerializer
 
+from dojo.models import FileUpload
+
 from . import utils
 
 # Reduce the scope of search for serializers.
 SERIALIZER_DEFS_MODULE = "dojo.api_v2.serializers"
 
+preferred_serializers = {
+    FileUpload: "FileSerializer",
+}
+
 
 class _Prefetcher:
     @staticmethod
@@ -31,7 +37,11 @@ def _is_model_serializer(obj):
 
         for _, serializer in available_serializers:
             model = serializer.Meta.model
-            serializers[model] = serializer
+            if model in preferred_serializers:
+                if serializer.__name__ == preferred_serializers[model]:
+                    serializers[model] = serializer
+            else:
+                serializers[model] = serializer
         # We add object->None to have a more uniform processing later on
         serializers[object] = None