Add support for using custom serializer for serializing faceted objects

drf_haystack/mixins.py

@@ -40,6 +40,7 @@ class FacetMixin(object):
 
     facet_filter_backends = [HaystackFacetFilter]
     facet_serializer_class = None
+    facet_objects_serializer_class = None
 
     @list_route(methods=["get"], url_path="facets")
     def facets(self, request):
@@ -101,3 +102,26 @@ def get_facet_serializer_class(self):
                 {"cls": self.__class__.__name__}
             )
         return self.facet_serializer_class
+
+    def get_facet_objects_serializer(self, *args, **kwargs):
+        """
+        Return the serializer instance which should be used for
+        serializing faceted objects.
+        """
+        assert "objects" in kwargs, "`objects` is a required argument to `get_facet_objects_serializer()`"
+
+        facet_objects_serializer_class = self.get_facet_objects_serializer_class()
+        kwargs["context"] = self.get_serializer_context()
+        kwargs["context"].update({
+            "objects": kwargs.pop("objects")
+        })
+        return facet_objects_serializer_class(*args, **kwargs)
+
+    def get_facet_objects_serializer_class(self):
+        """
+        Return the class to use for serializing faceted objects.
+        Defaults to using the views ``self.serializer_class`` if not
+        ``self.facet_objects_serializer_class`` is set.
+        """
+        return self.facet_objects_serializer_class or super(FacetMixin, self).get_serializer_class()
+

drf_haystack/serializers.py

@@ -401,7 +401,7 @@ def get_objects(self, instance):
 
         page = view.paginate_queryset(queryset)
         if page is not None:
-            serializer = view.get_serializer(page, many=True)
+            serializer = view.get_facet_objects_serializer(page, many=True)
             return OrderedDict([
                 ("count", _get_count(queryset)),
                 ("next", view.paginator.get_next_link()),