src/social/templatetags/social_tags.py

   1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 import re
   5 from django import template
   6 from django.utils.functional import lazy
   7 from django.utils.cache import add_never_cache_headers
   8 from catalogue.models import Book
   9 from ssify import ssi_variable
  10 from ssify.utils import ssi_vary_on_cookie
  11 from social.utils import likes, get_or_choose_cite
  12 from ..models import Carousel
  13
  14 register = template.Library()
  15
  16
  17 @ssi_variable(register, patch_response=[ssi_vary_on_cookie])
  18 def likes_book(request, book_id):
  19     return likes(request.user, Book.objects.get(pk=book_id), request)
  20
  21
  22 @ssi_variable(register, name='choose_cite', patch_response=[add_never_cache_headers])
  23 def choose_cite_tag(request, book_id=None, tag_ids=None):
  24     cite = get_or_choose_cite(request, book_id, tag_ids)
  25     return cite.pk if cite is not None else None
  26
  27
  28 @register.inclusion_tag('social/cite_promo.html')
  29 def render_cite(cite):
  30     return {
  31         'cite': cite,
  32     }
  33
  34
  35 @ssi_variable(register, patch_response=[ssi_vary_on_cookie])
  36 def book_shelf_tags(request, book_id):
  37     if not request.user.is_authenticated():
  38         return ''
  39     book = Book.objects.get(pk=book_id)
  40     lks = likes(request.user, book, request)
  41
  42     def get_value():
  43         if not lks:
  44             return ''
  45         tags = book.tags.filter(category='set', user=request.user).exclude(name='')
  46         if not tags:
  47             return ''
  48         ctx = {'tags': tags}
  49         return template.loader.render_to_string('social/shelf_tags.html', ctx)
  50     return lazy(get_value, str)()
  51
  52
  53 @register.inclusion_tag('social/carousel.html')
  54 def carousel(slug):
  55     # TODO: cache
  56     try:
  57         carousel = Carousel.objects.get(slug=slug)
  58     except Carousel.DoesNotExist:
  59         # TODO: add sanity check for install.
  60         carousel = None
  61     return {
  62         'carousel': carousel
  63     }
  64
  65
  66 @register.inclusion_tag('social/embed_video.html')
  67 def embed_video(url):
  68     m = re.match(r'https://www.youtube.com/watch\?v=([^&;]+)', url)
  69     return {
  70         'youtube_id': m.group(1) if m else None,
  71     }