Advertisement
Guest User

Untitled

a guest
Sep 19th, 2017
54
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.14 KB | None | 0 0
  1. from django.db.models import Q, Case, When, Value, IntegerField
  2.  
  3. class SearchResultsAdminMixin(object):
  4. def get_search_results(self, request, queryset, search_term):
  5. ''' Show exact match for title and slug at top of admin search results.
  6. '''
  7. qs, use_distinct = \
  8. super(SearchResultsAdminMixin, self).get_search_results(
  9. request, queryset, search_term)
  10.  
  11. search_term = search_term.strip()
  12. if not search_term:
  13. return qs, use_distinct
  14.  
  15. def cond_int(query):
  16. return Case(
  17. When(query, then=Value(1)),
  18. default=Value(0),
  19. output_field=IntegerField()
  20. )
  21.  
  22. qs = qs.annotate(
  23. exact_title=cond_int(Q(title__iexact=search_term)),
  24. exact_slug=cond_int(Q(slug__iexact=search_term)),
  25. )
  26.  
  27. order_by = []
  28. if qs.filter(exact_title=1).exists():
  29. order_by.append('-exact_title')
  30. if qs.filter(exact_slug=1).exists():
  31. order_by.append('-exact_slug')
  32.  
  33. if order_by:
  34. qs = qs.order_by(*order_by)
  35.  
  36. return qs, use_distinct
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement