2020-11-18 22:54:13 +00:00
|
|
|
from django.contrib import messages
|
2020-11-17 23:42:57 +00:00
|
|
|
|
2020-11-18 22:54:13 +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):
|
2020-11-18 22:54:13 +00:00
|
|
|
''' 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
|
2020-11-18 22:54:13 +00:00
|
|
|
try:
|
2020-11-19 23:19:33 +00:00
|
|
|
qsearch = query_search_index(search_term)
|
2020-11-18 22:54:13 +00:00
|
|
|
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
|
2020-11-18 22:54:13 +00:00
|
|
|
finally:
|
|
|
|
return qs, use_distinct
|