Use DB to augment ES hits with required info

WordPress · Nov 28, 2023 · b5fb362 · b5fb362
1 parent 14292b1
commit b5fb362
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/api/api/views/media_views.py b/api/api/views/media_views.py
@@ -267,6 +267,10 @@ def related(self, request, identifier=None, *_, **__):
 
         serializer_context = self.get_serializer_context()
 
+        serializer_class = self.get_serializer()
+        if serializer_class.needs_db:
+            results = self.get_db_results(results)
+
         serializer = self.get_serializer(results, many=True, context=serializer_context)
         return self.get_paginated_response(serializer.data)
 

diff --git a/api/test/media_integration.py b/api/test/media_integration.py
@@ -227,6 +227,10 @@ def get_terms_set(res):
             or result["creator"] == item["creator"]
         ), f"{terms_set} {get_terms_set(result)}/{result['creator']}-{item['creator']}"
 
+        assert result["license_version"] is not None
+        assert result["attribution"] is not None
+        assert result["creator_url"] is not None
+
 
 def sensitive_search_and_detail(media_type):
     search_res = requests.get(