index = MockIndex()
@staticmethod
- def _find_some_books(snippets=False, query_terms=None, max_results=20):
+ def _find_some_books(query_terms=None, max_results=20):
from .index import SearchResult
qs = Book.objects.order_by('?')
- if snippets:
- qs = qs.exclude(fragments=None)
results = []
for book in qs[:randint(1, max_results)]:
doc = {
'book_id': book.pk,
'published_date': randint(1000, 1920),
}
- if snippets:
- fragment = book.fragments.order_by('?')[0]
- doc.update({
- 'header_type': choice(['strofa', 'akap']),
- 'header_index': randint(100, 200),
- 'header_span': randint(100, 200),
- 'fragment_anchor': fragment.anchor,
- 'snippets_position': randint(100, 200),
- 'snippets_length': randint(100, 200),
- 'snippets_revision': randint(1, 100),
- 'themes_pl': fragment.tags.filter(category='theme'),
- })
res = SearchResult(doc, how_found='mock', query_terms=query_terms)
- if snippets:
- res.snippets = [fragment.short_text]
results.append(res)
return results
- def search_phrase(self, searched, field='text', book=False, filters=None, snippets=False):
- return self._find_some_books(snippets)
-
- def search_some(self, searched, fields, book=True, filters=None, snippets=True, query_terms=None):
- return self._find_some_books(snippets, query_terms)
-
- # WTF
- def search_books(self, query, filters=None, max_results=10):
- return self._find_some_books(snippets, max_results=max_results)
-
def search_everywhere(self, searched, query_terms=None):
return []
- def hint_tags(self, query, pdcounter=True, prefix=True):
- return Tag.objects.exclude(category='set').order_by('?')[:randint(1, 10)]
-
- def hint_books(self, prefix):
- return Book.objects.order_by('?')[:randint(1, 10)]