Ver código fonte

fix searching by url contents

Nick Sweeting 4 anos atrás
pai
commit
cadac4838e
2 arquivos alterados com 9 adições e 12 exclusões
  1. 8 11
      archivebox/core/mixins.py
  2. 1 1
      archivebox/main.py

+ 8 - 11
archivebox/core/mixins.py

@@ -2,22 +2,19 @@ from django.contrib import messages
 
 from archivebox.search import query_search_index
 
-class SearchResultsAdminMixin(object):
-    def get_search_results(self, request, queryset, search_term):
-        ''' Enhances the search queryset with results from the search backend.
-        '''
-        qs, use_distinct = \
-            super(SearchResultsAdminMixin, self).get_search_results(
-                request, queryset, search_term)
+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)
 
         search_term = search_term.strip()
         if not search_term:
             return qs, use_distinct
         try:
             qsearch = query_search_index(search_term)
+            qs = qs | qsearch
         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:
-            qs = queryset & qsearch
-        finally:
-            return qs, use_distinct
+        
+        return qs, use_distinct

+ 1 - 1
archivebox/main.py

@@ -448,7 +448,7 @@ def init(force: bool=False, quick: bool=False, setup: bool=False, out_dir: Path=
 
     if setup:
         run_subcommand('setup', pwd=out_dir)
-    
+
     if Snapshot.objects.count() < 25:     # hide the hints for experienced users
         print()
         print('    {lightred}Hint:{reset} To view your archive index, run:'.format(**ANSI))