b19593e0cf2b2b2d1a67de6b005c87e9fe028e26
[wolnelektury.git] / apps / search / templatetags / search_tags.py
1 # -*- coding: utf-8 -*-
2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 # import feedparser
6 # import datetime
7
8 from django import template
9 from django.template import Node, Variable
10 from django.utils.encoding import smart_str
11 from django.core.urlresolvers import reverse
12 # from django.contrib.auth.forms import UserCreationForm, AuthenticationForm
13 # from django.db.models import Q
14 from django.conf import settings
15 # from django.utils.translation import ugettext as _
16 from catalogue.templatetags.catalogue_tags import book_wide
17 from catalogue.models import Book
18 # from catalogue.forms import SearchForm
19 # from catalogue.utils import split_tags
20
21
22 register = template.Library()
23
24
25 @register.inclusion_tag('catalogue/book_searched.html', takes_context=True)
26 def book_searched(context, result):
27     book = Book.objects.get(pk=result.book_id)
28
29     # snippets = []
30     # for hit in result.hits:
31     #     if hit['snippets']:
32     #         snippets.append(hit['snippets'])
33     #     elif hit['fragment']:
34     #         snippets.append(hit['fragment'].short_text)
35
36     # We don't need hits which lead to sections but do not have
37     # snippets.
38     hits = filter(lambda (idx, h):
39                   result.snippets[idx] is not None
40                   or 'fragment' in h, enumerate(result.hits))
41     print "[tmpl: from %d hits selected %d]" % (len(result.hits), len(hits))
42
43     for (idx, hit) in hits:
44         # currently we generate one snipper per hit though.
45         if 'fragment' in hit:
46             continue
47         snip = result.snippets[idx]
48         # fix some formattting
49         snip = snip.replace("\n", "<br />").replace('---', '&mdash;')
50         hit['snippet'] = snip
51
52     return {
53         'related': book.related_info(),
54         'book': book,
55         'main_link': book.get_absolute_url(),
56         'request': context.get('request'),
57         'hits': zip(*hits)[1],
58         'main_link': book.get_absolute_url(),
59     }