src/search/views.py

   1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 from django.conf import settings
   5 from django.shortcuts import render
   6 from django.views.decorators import cache
   7 from django.http import HttpResponse, JsonResponse
   8
   9 from catalogue.models import Book, Tag
  10 from .forms import SearchFilters
  11 import re
  12 import json
  13
  14 from wolnelektury.utils import re_escape
  15
  16
  17 query_syntax_chars = re.compile(r"[\\/*:(){}?.[\]+]")
  18
  19
  20 def remove_query_syntax_chars(query, replace=' '):
  21     return query_syntax_chars.sub(replace, query)
  22
  23
  24 @cache.never_cache
  25 def hint(request, mozhint=False, param='term'):
  26     prefix = request.GET.get(param, '')
  27     if len(prefix) < 2:
  28         return JsonResponse([], safe=False)
  29
  30     prefix = re_escape(' '.join(remove_query_syntax_chars(prefix).split()))
  31
  32     try:
  33         limit = int(request.GET.get('max', ''))
  34     except ValueError:
  35         limit = 20
  36     else:
  37         if limit < 1:
  38             limit = 20
  39
  40     authors = Tag.objects.filter(
  41         category='author', name_pl__iregex='\m' + prefix).only('name', 'id', 'slug', 'category')
  42     data = [
  43         {
  44             'label': author.name,
  45             'id': author.id,
  46             'url': author.get_absolute_url(),
  47         }
  48         for author in authors[:limit]
  49     ]
  50     if len(data) < limit:
  51         for b in Book.objects.filter(findable=True, title__iregex='\m' + prefix)[:limit-len(data)]:
  52             author_str = b.author_unicode()
  53             translator = b.translator()
  54             if translator:
  55                 author_str += ' (tłum. ' + translator + ')'
  56             data.append(
  57                 {
  58                     'label': b.title,
  59                     'author': author_str,
  60                     'id': b.id,
  61                     'url': b.get_absolute_url()
  62                 }
  63             )
  64
  65     if mozhint:
  66         data = [
  67             prefix,
  68             [
  69                 item['label']
  70                 for item in data
  71             ]
  72         ]
  73
  74     callback = request.GET.get('callback', None)
  75     if callback:
  76         return HttpResponse("%s(%s);" % (callback, json.dumps(data)),
  77                             content_type="application/json; charset=utf-8")
  78     else:
  79         return JsonResponse(data, safe=False)
  80
  81
  82
  83 @cache.never_cache
  84 def search(request):
  85     filters = SearchFilters(request.GET)
  86     ctx = {
  87         'title': 'Wynik wyszukiwania',
  88         'query': request.GET.get('q', ''),
  89         'filters': filters,
  90     }
  91     if filters.is_valid():
  92         ctx['results'] = filters.results()
  93         for k, v in ctx['results'].items():
  94             if v:
  95                 ctx['hasresults'] = True
  96                 break
  97     return render(request, 'search/results.html', ctx)