apps/search/views.py

   1 # -*- coding: utf-8 -*-
   2
   3 from django.conf import settings
   4 from django.shortcuts import render_to_response, get_object_or_404
   5 from django.template import RequestContext
   6 from django.contrib.auth.decorators import login_required
   7 from django.views.decorators import cache
   8 from django.http import HttpResponse, HttpResponseRedirect, Http404, HttpResponsePermanentRedirect
   9 from django.utils.translation import ugettext as _
  10
  11 from catalogue.utils import get_random_hash
  12 from catalogue.models import Book, Tag, Fragment
  13 from catalogue.fields import dumps
  14 from catalogue.views import JSONResponse
  15 from search import Search, JVM, SearchResult
  16 from lucene import StringReader
  17 from suggest.forms import PublishingSuggestForm
  18
  19 import enchant
  20
  21 dictionary = enchant.Dict('pl_PL')
  22
  23
  24 def match_word_re(word):
  25     if 'sqlite' in settings.DATABASES['default']['ENGINE']:
  26         return r"\b%s\b" % word
  27     elif 'mysql' in settings.DATABASES['default']['ENGINE']:
  28         return "[[:<:]]%s[[:>:]]" % word
  29
  30
  31 def did_you_mean(query, tokens):
  32     change = {}
  33     for t in tokens:
  34         authors = Tag.objects.filter(category='author', name__iregex=match_word_re(t))
  35         if len(authors) > 0:
  36             continue
  37
  38         if not dictionary.check(t):
  39             try:
  40                 change[t] = dictionary.suggest(t)[0]
  41             except IndexError:
  42                 pass
  43
  44     if change == {}:
  45         return None
  46
  47     for frm, to in change.items():
  48         query = query.replace(frm, to)
  49
  50     return query
  51
  52
  53 def hint(request):
  54     prefix = request.GET.get('term', '')
  55     if len(prefix) < 2:
  56         return JSONResponse([])
  57     JVM.attachCurrentThread()
  58     s = Search()
  59
  60     hint = s.hint()
  61     try:
  62         tags = request.GET.get('tags', '')
  63         hint.tags(Tag.get_tag_list(tags))
  64     except:
  65         pass
  66
  67     # tagi beda ograniczac tutaj
  68     # ale tagi moga byc na ksiazce i na fragmentach
  69     # jezeli tagi dot tylko ksiazki, to wazne zeby te nowe byly w tej samej ksiazce
  70     # jesli zas dotycza themes, to wazne, zeby byly w tym samym fragmencie.
  71
  72     tags = s.hint_tags(prefix)
  73     books = s.hint_books(prefix)
  74
  75     # TODO DODAC TU HINTY
  76
  77     return JSONResponse(
  78         [{'label': t.name,
  79           'category': _(t.category),
  80           'id': t.id,
  81           'url': t.get_absolute_url()}
  82           for t in tags] + \
  83           [{'label': b.title,
  84             'category': _('book'),
  85             'id': b.id,
  86             'url': b.get_absolute_url()}
  87             for b in books])
  88
  89
  90 def main(request):
  91     results = {}
  92     JVM.attachCurrentThread()  # where to put this?
  93     srch = Search()
  94
  95     results = None
  96     query = None
  97     fuzzy = False
  98
  99     if 'q' in request.GET:
 100         tags = request.GET.get('tags', '')
 101         query = request.GET['q']
 102         book_id = request.GET.get('book', None)
 103         book = None
 104         if book_id is not None:
 105             book = get_object_or_404(Book, id=book_id)
 106
 107         hint = srch.hint()
 108         try:
 109             tag_list = Tag.get_tag_list(tags)
 110         except:
 111             tag_list = []
 112
 113         if len(query) < 2:
 114             return render_to_response('catalogue/search_too_short.html', {'tags': tag_list, 'prefix': query},
 115                                       context_instance=RequestContext(request))
 116
 117         hint.tags(tag_list)
 118         if book:
 119             hint.books(book)
 120
 121         toks = StringReader(query)
 122         fuzzy = 'fuzzy' in request.GET
 123         if fuzzy:
 124             fuzzy = 0.7
 125
 126         results = SearchResult.aggregate(srch.search_perfect_book(toks, fuzzy=fuzzy, hint=hint),
 127                                          srch.search_book(toks, fuzzy=fuzzy, hint=hint),
 128                                          srch.search_perfect_parts(toks, fuzzy=fuzzy, hint=hint),
 129                                          srch.search_everywhere(toks, fuzzy=fuzzy, hint=hint))
 130
 131         for r in results:
 132             r.process_hits()
 133
 134         results.sort(reverse=True)
 135
 136         for r in results:
 137             print "-----"
 138             for h in r.hits:
 139                 print "- %s" % h
 140
 141                 # Did you mean?
 142         suggestion = did_you_mean(query, srch.get_tokens(toks, field="SIMPLE"))
 143
 144         if len(results) == 1:
 145             if len(results[0].hits) == 0:
 146                 return HttpResponseRedirect(results[0].book.get_absolute_url())
 147             elif len(results[0].hits) == 1 and results[0].hits[0] is not None:
 148                 frag = Fragment.objects.get(anchor=results[0].hits[0])
 149                 return HttpResponseRedirect(frag.get_absolute_url())
 150         elif len(results) == 0:
 151             form = PublishingSuggestForm(initial={"books": query + ", "})
 152             return render_to_response('catalogue/search_no_hits.html',
 153                                       {'tags': tag_list,
 154                                        'prefix': query,
 155                                        "form": form,
 156                                        'did_you_mean': suggestion},
 157                 context_instance=RequestContext(request))
 158
 159         return render_to_response('catalogue/search_multiple_hits.html',
 160                                   {'tags': tag_list,
 161                                    'prefix': query,
 162                                    'results': results,
 163                                    'did_you_mean': suggestion},
 164             context_instance=RequestContext(request))