Add Book.ancestor m2m.
[wolnelektury.git] / apps / social / templatetags / social_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 from random import randint
6 from django.db.models import Q
7 from django import template
8 from catalogue.models import Book
9 from social.models import Cite
10 from social.utils import likes, cites_for_tags
11
12 register = template.Library()
13
14 register.filter('likes', likes)
15
16
17 @register.assignment_tag(takes_context=True)
18 def choose_cite(context, ctx=None):
19     """Choose a cite for main page, for book or for set of tags."""
20     try:
21         request = context['request']
22         assert request.user.is_staff
23         assert 'choose_cite' in request.GET
24         cite = Cite.objects.get(pk=request.GET['choose_cite'])
25     except (AssertionError, Cite.DoesNotExist):
26         if ctx is None:
27             cites = Cite.objects.all()
28         elif isinstance(ctx, Book):
29             cites = Cite.objects.filter(Q(book=ctx) | Q(book__ancestor=ctx))
30         else:
31             cites = cites_for_tags(ctx)
32         stickies = cites.filter(sticky=True)
33         count = stickies.count()
34         if count:
35             cite = stickies[randint(0, count - 1)]
36         else:
37             count = cites.count()
38             if count:
39                 cite = cites[randint(0, count - 1)]
40             else:
41                 cite = None
42     return cite
43
44
45 @register.inclusion_tag('social/cite_promo.html')
46 def render_cite(cite):
47     return {
48         'cite': cite,
49     }
50
51
52 @register.inclusion_tag('social/cite_promo.html', takes_context=True)
53 def cite_promo(context, ctx=None, fallback=False):
54     return {
55         'cite': choose_cite(context, ctx),
56         'fallback': fallback,
57         'ctx': ctx,
58     }
59
60
61 @register.inclusion_tag('social/shelf_tags.html', takes_context=True)
62 def shelf_tags(context, book):
63     user = context['request'].user
64     if not user.is_authenticated():
65         tags = []
66     else:
67         tags = book.tags.filter(category='set', user=user).exclude(name='')
68     return {'tags': tags}