Skip to content

Commit

Permalink
Fix SearchFilter.must_call_distinict for annotation+m2m (#7146)
Browse files Browse the repository at this point in the history
* Test SearchFilter annotation+m2m distinct

* Fix SearchFilter annotation+m2m distinct
  • Loading branch information
rpkilby committed May 11, 2020
1 parent 4ac0fae commit 8a38991
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 1 deletion.
2 changes: 1 addition & 1 deletion rest_framework/filters.py
Expand Up @@ -85,7 +85,7 @@ def must_call_distinct(self, queryset, search_fields):
search_field = search_field[1:]
# Annotated fields do not need to be distinct
if isinstance(queryset, models.QuerySet) and search_field in queryset.query.annotations:
return False
continue
parts = search_field.split(LOOKUP_SEP)
for part in parts:
field = opts.get_field(part)
Expand Down
15 changes: 15 additions & 0 deletions tests/test_filters.py
Expand Up @@ -406,6 +406,21 @@ class SearchListView(generics.ListAPIView):
assert len(response.data) == 1
assert response.data[0]['title_text'] == 'ABCDEF'

def test_must_call_distinct_subsequent_m2m_fields(self):
f = filters.SearchFilter()

queryset = SearchFilterModelM2M.objects.annotate(
title_text=Upper(
Concat(models.F('title'), models.F('text'))
)
).all()

# Sanity check that m2m must call distinct
assert f.must_call_distinct(queryset, ['attributes'])

# Annotated field should not prevent m2m must call distinct
assert f.must_call_distinct(queryset, ['title_text', 'attributes'])


class OrderingFilterModel(models.Model):
title = models.CharField(max_length=20, verbose_name='verbose title')
Expand Down

0 comments on commit 8a38991

Please sign in to comment.