wagtail_bakerydemo/bakerydemo/search/views.py

from django.shortcuts import render
from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger

from wagtail.wagtailcore.models import Page
from wagtail.wagtailsearch.models import Query

from bakerydemo.blog.models import BlogPage
from bakerydemo.breads.models import BreadPage
from bakerydemo.locations.models import LocationPage
from bakerydemo.base.models import People


def search(request):
    # Search
    search_query = request.GET.get('query', None)
    if search_query:
        '''
        Because we can't use ElasticSearch for the demo, we use the native db search.
        But native DB search can't search specific fields in our models on a `Page` query.
        So for demo purposes ONLY, we hard-code in the model names we want to search.
        In production, use ElasticSearch and a simplified search query, per
        http://docs.wagtail.io/en/v1.8.1/topics/search/searching.html
        '''

        blog_results = BlogPage.objects.live().search(search_query)
        blog_page_ids = [p.page_ptr.id for p in blog_results]

        bread_results = BreadPage.objects.live().search(search_query)
        bread_page_ids = [p.page_ptr.id for p in bread_results]

        location_results = LocationPage.objects.live().search(search_query)
        location_result_ids = [p.page_ptr.id for p in location_results]

        page_ids = blog_page_ids + bread_page_ids + location_result_ids
        search_results = Page.objects.live().filter(id__in=page_ids)

        query = Query.get(search_query)

        # Record hit
        query.add_hit()

    else:
        search_results = Page.objects.none()

    # Pagination
    page = request.GET.get('page', 1)
    paginator = Paginator(search_results, 10)
    try:
        search_results = paginator.page(page)
    except PageNotAnInteger:
        search_results = paginator.page(1)
    except EmptyPage:
        search_results = paginator.page(paginator.num_pages)

    return render(request, 'search/search_results.html', {
        'search_query': search_query,
        'search_results': search_results,
    })