apps/search/views.py

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