1
0
Fork 0
mirror of synced 2024-07-01 20:40:41 +12:00
ArchiveBox/archivebox/core/mixins.py

22 lines
897 B
Python
Raw Normal View History

from django.contrib import messages
2020-11-18 12:42:57 +13:00
from archivebox.search import query_search_index
2020-11-18 12:42:57 +13:00
2021-04-06 15:21:07 +12:00
class SearchResultsAdminMixin:
def get_search_results(self, request, queryset, search_term: str):
"""Enhances the search queryset with results from the search backend"""
qs, use_distinct = super().get_search_results(request, queryset, search_term)
2020-11-18 12:42:57 +13:00
search_term = search_term.strip()
if not search_term:
return qs, use_distinct
try:
qsearch = query_search_index(search_term)
2021-04-06 15:21:07 +12:00
qs = qs | qsearch
except Exception as err:
print(f'[!] Error while using search backend: {err.__class__.__name__} {err}')
messages.add_message(request, messages.WARNING, f'Error from the search backend, only showing results from default admin search fields - Error: {err}')
2021-04-06 15:21:07 +12:00
return qs, use_distinct