e5ea59837b5f6d4c12e78c9eb1074bef2a715c2d
[wolnelektury.git] / 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)