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)