From cefd547c611a330f9736b8cc98166a3a4c1d739d Mon Sep 17 00:00:00 2001
From: ch3nnn <chenyinren@vip.qq.com>
Date: Thu, 11 Aug 2022 20:35:26 +0800
Subject: [PATCH 1/2] =?UTF-8?q?feat:=20=E6=96=B0=E5=A2=9E=20elasticsearch?=
 =?UTF-8?q?=20=E6=90=9C=E7=B4=A2=20suggest=5Fsearch=20=E6=8B=BC=E5=86=99?=
 =?UTF-8?q?=E7=BA=A0=E6=AD=A3=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 blog/views.py                       | 22 +++++++++++--
 djangoblog/elasticsearch_backend.py | 50 +++++++++++++++++++++++++++--
 djangoblog/urls.py                  | 12 ++++---
 templates/search/search.html        | 12 +++++--
 4 files changed, 85 insertions(+), 11 deletions(-)

diff --git a/blog/views.py b/blog/views.py
index c9ff116..6cc05cd 100644
--- a/blog/views.py
+++ b/blog/views.py
@@ -11,10 +11,11 @@ from django.shortcuts import render
 from django.views.decorators.csrf import csrf_exempt
 from django.views.generic.detail import DetailView
 from django.views.generic.list import ListView
+from haystack.views import SearchView
 
-from blog.models import Article, Category, Tag, Links, LinkShowType
+from blog.models import Article, Category, LinkShowType, Links, Tag
 from comments.forms import CommentForm
-from djangoblog.utils import cache, get_sha256, get_blog_setting
+from djangoblog.utils import cache, get_blog_setting, get_sha256
 
 logger = logging.getLogger(__name__)
 
@@ -267,6 +268,23 @@ class LinkListView(ListView):
         return Links.objects.filter(is_enable=True)
 
 
+class EsSearchView(SearchView):
+    def get_context(self):
+        paginator, page = self.build_page()
+        context = {
+            "query": self.query,
+            "form": self.form,
+            "page": page,
+            "paginator": paginator,
+            "suggestion": None,
+        }
+        if hasattr(self.results, "query") and self.results.query.backend.include_spelling:
+            context["suggestion"] = self.results.query.get_spelling_suggestion()
+        context.update(self.extra_context())
+
+        return context
+
+
 @csrf_exempt
 def fileupload(request):
     """
diff --git a/djangoblog/elasticsearch_backend.py b/djangoblog/elasticsearch_backend.py
index 5f875f4..eb7fea2 100644
--- a/djangoblog/elasticsearch_backend.py
+++ b/djangoblog/elasticsearch_backend.py
@@ -1,6 +1,7 @@
 from django.utils.encoding import force_str
 from elasticsearch_dsl import Q
 from haystack.backends import BaseEngine, BaseSearchBackend, BaseSearchQuery, log_query
+from haystack.forms import ModelSearchForm
 from haystack.models import SearchResult
 from haystack.utils import log as logging
 
@@ -18,6 +19,7 @@ class ElasticSearchBackend(BaseSearchBackend):
             connection_alias,
             **connection_options)
         self.manager = ArticleDocumentManager()
+        self.include_spelling = True
 
     def _get_models(self, iterable):
         models = iterable if iterable and iterable[0] else Article.objects.all()
@@ -51,6 +53,21 @@ class ElasticSearchBackend(BaseSearchBackend):
     def clear(self, models=None, commit=True):
         self.remove(None)
 
+    @staticmethod
+    def get_suggestion(body: str):
+        """获取建议 keyword """
+        search = ArticleDocument.search() \
+            .query("match", body=body) \
+            .suggest('suggest_search', body, term={'field': 'body'}) \
+            .execute()
+
+        keywords = []
+        for suggest in search.suggest.suggest_search:
+            if suggest["options"]:
+                keywords.append(suggest["options"][0]["text"])
+                
+        return ' '.join(keywords) if keywords else body
+
     @log_query
     def search(self, query_string, **kwargs):
         logger.info('search query_string:' + query_string)
@@ -58,8 +75,16 @@ class ElasticSearchBackend(BaseSearchBackend):
         start_offset = kwargs.get('start_offset')
         end_offset = kwargs.get('end_offset')
 
-        q = Q('bool', should=[Q('match', body=query_string), Q(
-            'match', title=query_string)], minimum_should_match="70%")
+        # 搜索建议
+        is_suggest = getattr(self, "is_suggest", None)
+        if is_suggest is not False:
+            suggestion = self.get_suggestion(query_string)
+        else:
+            suggestion = query_string
+
+        q = Q('bool',
+              should=[Q('match', body=suggestion), Q('match', title=suggestion)],
+              minimum_should_match="70%")
 
         search = ArticleDocument.search() \
                      .query('bool', filter=[q]) \
@@ -85,7 +110,7 @@ class ElasticSearchBackend(BaseSearchBackend):
                 **additional_fields)
             raw_results.append(result)
         facets = {}
-        spelling_suggestion = None
+        spelling_suggestion = None if query_string == suggestion else suggestion
 
         return {
             'results': raw_results,
@@ -134,6 +159,25 @@ class ElasticSearchQuery(BaseSearchQuery):
         results = self.get_results()
         return len(results) if results else 0
 
+    def get_spelling_suggestion(self, preferred_query=None):
+        return self._spelling_suggestion
+
+    def build_params(self, spelling_query=None):
+        kwargs = super(ElasticSearchQuery, self).build_params(spelling_query=spelling_query)
+        return kwargs
+
+
+class ElasticSearchModelSearchForm(ModelSearchForm):
+
+    def search(self):
+        # 是否建议搜索
+        self.searchqueryset.query.backend.is_suggest = True
+        if self.data.get("is_suggest") == "no":
+            self.searchqueryset.query.backend.is_suggest = False
+
+        sqs = super().search()
+        return sqs
+
 
 class ElasticSearchEngine(BaseEngine):
     backend = ElasticSearchBackend
diff --git a/djangoblog/urls.py b/djangoblog/urls.py
index 6d88345..a834684 100644
--- a/djangoblog/urls.py
+++ b/djangoblog/urls.py
@@ -14,14 +14,17 @@ Including another URLconf
     2. Add a URL to urlpatterns:  url(r'^blog/', include('blog.urls'))
 """
 from django.conf import settings
-from django.urls import include, re_path
 from django.conf.urls.static import static
 from django.contrib.sitemaps.views import sitemap
 from django.urls import include
+from django.urls import re_path
+from haystack.views import search_view_factory
 
+from blog.views import EsSearchView
 from djangoblog.admin_site import admin_site
+from djangoblog.elasticsearch_backend import ElasticSearchModelSearchForm
 from djangoblog.feeds import DjangoBlogFeed
-from djangoblog.sitemap import StaticViewSitemap, ArticleSiteMap, CategorySiteMap, TagSiteMap, UserSiteMap
+from djangoblog.sitemap import ArticleSiteMap, CategorySiteMap, StaticViewSitemap, TagSiteMap, UserSiteMap
 
 sitemaps = {
 
@@ -43,10 +46,11 @@ urlpatterns = [
                   re_path(r'', include('accounts.urls', namespace='account')),
                   re_path(r'', include('oauth.urls', namespace='oauth')),
                   re_path(r'^sitemap\.xml$', sitemap, {'sitemaps': sitemaps},
-                      name='django.contrib.sitemaps.views.sitemap'),
+                          name='django.contrib.sitemaps.views.sitemap'),
                   re_path(r'^feed/$', DjangoBlogFeed()),
                   re_path(r'^rss/$', DjangoBlogFeed()),
-                  re_path(r'^search', include('haystack.urls'), name='search'),
+                  re_path('^search', search_view_factory(view_class=EsSearchView, form_class=ElasticSearchModelSearchForm),
+                          name='search'),
                   re_path(r'', include('servermanager.urls', namespace='servermanager')),
                   re_path(r'', include('owntracks.urls', namespace='owntracks'))
               ] + static(settings.STATIC_URL, document_root=settings.STATIC_ROOT)
diff --git a/templates/search/search.html b/templates/search/search.html
index 128aef4..1404c60 100644
--- a/templates/search/search.html
+++ b/templates/search/search.html
@@ -15,8 +15,16 @@
         <div id="content" role="main">
             {% if query %}
                 <header class="archive-header">
-
-                    <h2 class="archive-title"> 搜索：<span style="color: red">{{ query }}</span></h2>
+                    {% if suggestion %}
+                        <h2 class="archive-title">
+                            已显示<span style="color: red"> “{{ suggestion }}” </span>的搜索结果。&nbsp;&nbsp;
+                            仍然搜索：<a style="text-transform: none;" href="/search/?q={{ query }}&is_suggest=no">{{ query }}</a> <br>
+                        </h2>
+                    {% else %}
+                        <h2 class="archive-title">
+                            搜索：<span style="color: red">{{ query }} </span> &nbsp;&nbsp;
+                        </h2>
+                    {% endif %}
                 </header><!-- .archive-header -->
             {% endif %}
             {% if query and page.object_list %}

From 86029562c5a2ebb5cb593a0a66f7b731dc8c1181 Mon Sep 17 00:00:00 2001
From: ch3nnn <chenyinren@vip.qq.com>
Date: Wed, 17 Aug 2022 23:57:05 +0800
Subject: [PATCH 2/2] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96=20suggest=5Fsear?=
 =?UTF-8?q?ch=20=E6=8E=A8=E8=8D=90=E8=AF=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 djangoblog/elasticsearch_backend.py | 25 ++++++++++++-------------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/djangoblog/elasticsearch_backend.py b/djangoblog/elasticsearch_backend.py
index eb7fea2..4afe498 100644
--- a/djangoblog/elasticsearch_backend.py
+++ b/djangoblog/elasticsearch_backend.py
@@ -54,19 +54,22 @@ class ElasticSearchBackend(BaseSearchBackend):
         self.remove(None)
 
     @staticmethod
-    def get_suggestion(body: str):
-        """获取建议 keyword """
+    def get_suggestion(query: str) -> str:
+        """获取推荐词, 如果没有找到添加原搜索词"""
+
         search = ArticleDocument.search() \
-            .query("match", body=body) \
-            .suggest('suggest_search', body, term={'field': 'body'}) \
+            .query("match", body=query) \
+            .suggest('suggest_search', query, term={'field': 'body'}) \
             .execute()
 
         keywords = []
         for suggest in search.suggest.suggest_search:
             if suggest["options"]:
                 keywords.append(suggest["options"][0]["text"])
-                
-        return ' '.join(keywords) if keywords else body
+            else:
+                keywords.append(suggest["text"])
+
+        return ' '.join(keywords)
 
     @log_query
     def search(self, query_string, **kwargs):
@@ -75,9 +78,8 @@ class ElasticSearchBackend(BaseSearchBackend):
         start_offset = kwargs.get('start_offset')
         end_offset = kwargs.get('end_offset')
 
-        # 搜索建议
-        is_suggest = getattr(self, "is_suggest", None)
-        if is_suggest is not False:
+        # 推荐词搜索
+        if getattr(self, "is_suggest", None):
             suggestion = self.get_suggestion(query_string)
         else:
             suggestion = query_string
@@ -171,10 +173,7 @@ class ElasticSearchModelSearchForm(ModelSearchForm):
 
     def search(self):
         # 是否建议搜索
-        self.searchqueryset.query.backend.is_suggest = True
-        if self.data.get("is_suggest") == "no":
-            self.searchqueryset.query.backend.is_suggest = False
-
+        self.searchqueryset.query.backend.is_suggest = self.data.get("is_suggest") != "no"
         sqs = super().search()
         return sqs