Allows searching with SearchFilter in annotated fields

Fixes #6094

Author: andrzej3393

rest_framework/filters.py

@@ -79,6 +79,9 @@ def must_call_distinct(self, queryset, search_fields):
                 search_field = search_field[1:]
             parts = search_field.split(LOOKUP_SEP)
             for part in parts:
+                if isinstance(queryset, models.QuerySet) and part in queryset.query.annotations:
+                    # This field is annotated
+                    continue
                 field = opts.get_field(part)
                 if hasattr(field, 'get_path_info'):
                     # This field is a relation, update opts to follow the relation

tests/test_filters.py

@@ -5,6 +5,7 @@
 import pytest
 from django.core.exceptions import ImproperlyConfigured
 from django.db import models
+from django.db.models.functions import Concat
 from django.test import TestCase
 from django.test.utils import override_settings
 from django.utils.six.moves import reload_module
@@ -304,6 +305,35 @@ class SearchListView(generics.ListAPIView):
         assert len(response.data) == 1
 
 
+class SearchFilterAnnotatedSerializer(serializers.ModelSerializer):
+    title_text = serializers.CharField()
+
+    class Meta:
+        model = SearchFilterModel
+        fields = ('title', 'text', 'title_text')
+
+
+class SearchFilterAnnotatedFieldTests(TestCase):
+    @classmethod
+    def setUpTestData(cls):
+        SearchFilterModel.objects.create(title='abc', text='def')
+        SearchFilterModel.objects.create(title='ghi', text='jkl')
+
+    def test_search_in_annotated_field(self):
+        class SearchListView(generics.ListAPIView):
+            queryset = SearchFilterModel.objects.annotate(
+                title_text=Concat(models.F('title'), models.F('text'), output_field=models.CharField())).all()
+            serializer_class = SearchFilterAnnotatedSerializer
+            filter_backends = (filters.SearchFilter,)
+            search_fields = ('title_text',)
+
+        view = SearchListView.as_view()
+        request = factory.get('/', {'search': 'abcdef'})
+        response = view(request)
+        assert len(response.data) == 1
+        assert response.data[0]['title_text'] == 'abcdef'
+
+
 class OrderingFilterModel(models.Model):
     title = models.CharField(max_length=20, verbose_name='verbose title')
     text = models.CharField(max_length=100)