Remove picture models.
[wolnelektury.git] / src / search / views.py
1 # This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Wolne Lektury. 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 from sorl.thumbnail import get_thumbnail
9
10 import catalogue.models
11 import infopages.models
12 from .forms import SearchFilters
13 import re
14 import json
15
16 from wolnelektury.utils import re_escape
17
18
19 query_syntax_chars = re.compile(r"[\\/*:(){}?.[\]+]")
20
21
22 def remove_query_syntax_chars(query, replace=' '):
23     return query_syntax_chars.sub(replace, query)
24
25
26 @cache.never_cache
27 def hint(request, mozhint=False, param='term'):
28     prefix = request.GET.get(param, '')
29     if len(prefix) < 2:
30         return JsonResponse([], safe=False)
31
32     prefix = re_escape(' '.join(remove_query_syntax_chars(prefix).split()))
33
34     try:
35         limit = int(request.GET.get('max', ''))
36     except ValueError:
37         limit = 20
38     else:
39         if limit < 1:
40             limit = 20
41
42     data = []
43     if len(data) < limit:
44         authors = catalogue.models.Tag.objects.filter(
45             category='author', name_pl__iregex='\m' + prefix).only('name', 'id', 'slug', 'category')
46         data.extend([
47             {
48                 'type': 'author',
49                 'label': author.name,
50                 'url': author.get_absolute_url(),
51                 'img': get_thumbnail(author.photo, '72x72', crop='top').url if author.photo else '',
52             }
53             for author in authors[:limit - len(data)]
54         ])
55     if request.user.is_authenticated and len(data) < limit:
56         tags = catalogue.models.Tag.objects.filter(
57             category='set', user=request.user, name_pl__iregex='\m' + prefix).only('name', 'id', 'slug', 'category')
58         data.extend([
59             {
60                 'type': 'set',
61                 'label': tag.name,
62                 'url': tag.get_absolute_url(),
63             }
64             for tag in tags[:limit - len(data)]
65         ])
66     if len(data) < limit:
67         tags = catalogue.models.Tag.objects.filter(
68             category__in=('theme', 'genre', 'epoch', 'kind'), name_pl__iregex='\m' + prefix).only('name', 'id', 'slug', 'category')
69         data.extend([
70             {
71                 'type': tag.category,
72                 'label': tag.name,
73                 'url': tag.get_absolute_url(),
74             }
75             for tag in tags[:limit - len(data)]
76         ])
77     if len(data) < limit:
78         collections = catalogue.models.Collection.objects.filter(
79             title_pl__iregex='\m' + prefix).only('title', 'slug')
80         data.extend([
81             {
82                 'type': 'collection',
83                 'label': collection.title,
84                 'url': collection.get_absolute_url(),
85             }
86             for collection in collections[:limit - len(data)]
87         ])
88     if len(data) < limit:
89         for b in catalogue.models.Book.objects.filter(findable=True, title__iregex='\m' + prefix)[:limit-len(data)]:
90             author_str = b.author_unicode()
91             translator = b.translator()
92             if translator:
93                 author_str += ' (tłum. ' + translator + ')'
94             data.append(
95                 {
96                     'type': 'book',
97                     'label': b.title,
98                     'author': author_str,
99                     'url': b.get_absolute_url(),
100                     'img': get_thumbnail(b.cover_clean, '72x72').url if b.cover_clean else '',
101                 }
102             )
103     if len(data) < limit:
104         infos = infopages.models.InfoPage.objects.filter(
105             published=True,
106             findable=True,
107             title_pl__iregex='\m' + prefix).only('title', 'id', 'slug')
108         data.extend([
109             {
110                 'type': 'info',
111                 'label': info.title,
112                 'url': info.get_absolute_url(),
113             }
114             for info in infos[:limit - len(data)]
115         ])
116
117     if mozhint:
118         data = [
119             prefix,
120             [
121                 item['label']
122                 for item in data
123             ],
124             [
125                 item.get('author', '')
126                 for item in data
127             ],
128             [
129                 item['url']
130                 for item in data
131             ]
132         ]
133
134     callback = request.GET.get('callback', None)
135     if callback:
136         return HttpResponse("%s(%s);" % (callback, json.dumps(data)),
137                             content_type="application/json; charset=utf-8")
138     else:
139         return JsonResponse(data, safe=False)
140
141
142
143 @cache.never_cache
144 def search(request):
145     filters = SearchFilters(request.GET)
146     ctx = {
147         'title': 'Wynik wyszukiwania',
148         'query': request.GET.get('q', ''),
149         'filters': filters,
150     }
151     if filters.is_valid():
152         ctx['results'] = filters.results()
153         for k, v in ctx['results'].items():
154             if v:
155                 ctx['hasresults'] = True
156                 break
157     return render(request, 'search/results.html', ctx)