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}