ArchiveBox/archivebox/core/mixins.py

24 lines
903 B
Python
Raw Normal View History

from django.contrib import messages
2020-11-17 23:42:57 +00:00
from archivebox.search import query_search_index
2020-11-17 23:42:57 +00:00
class SearchResultsAdminMixin(object):
def get_search_results(self, request, queryset, search_term):
''' Enhances the search queryset with results from the search backend.
2020-11-17 23:42:57 +00:00
'''
qs, use_distinct = \
super(SearchResultsAdminMixin, self).get_search_results(
request, queryset, search_term)
search_term = search_term.strip()
if not search_term:
return qs, use_distinct
try:
qsearch = query_search_index(search_term)
except Exception as err:
messages.add_message(request, messages.WARNING, f'Error from the search backend, only showing results from default admin search fields - Error: {err}')
else:
2020-12-14 18:40:38 +00:00
qs = queryset & qsearch
finally:
return qs, use_distinct