X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/6280673f35e13e75e81c5b7821bd2a44a1831eab..357027375ff8867f42ca34bcbfb5a78b5b185fc3:/src/search/mock_search.py

diff --git a/src/search/mock_search.py b/src/search/mock_search.py
new file mode 100644
index 000000000..253c277a8
--- /dev/null
+++ b/src/search/mock_search.py
@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
+from mock import Mock
+from catalogue.models import Book, Tag
+from random import randint, choice
+
+
+class Search(Mock):
+    """
+    Search mock for development without setting up Solr.
+
+    Instead of connecting to an actual search server, it returns
+    some random results for any query.
+    """
+    class MockIndex(Mock):
+        def analyze(*args, **kwargs):
+            return []
+
+    index = MockIndex()
+
+    @staticmethod
+    def _find_some_books(snippets=False, 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 = {
+                'score': randint(0, 100),
+                '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)
+
+    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)]