src/social/templatetags/social_tags.py

   1 # This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Wolne Lektury. 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, Fragment
   9 from social.utils import likes, get_or_choose_cite, choose_cite as cs
  10 from ..models import Carousel, Cite
  11
  12 register = template.Library()
  13
  14
  15 @register.simple_tag(takes_context=True)
  16 def likes_book(context, book):
  17     request = context['request']
  18     return likes(request.user, book, request)
  19
  20
  21 @register.simple_tag(takes_context=True)
  22 def choose_cite(context, book_id=None, tag_ids=None):
  23     request = context['request']
  24     return get_or_choose_cite(request, book_id, tag_ids)
  25
  26
  27 @register.simple_tag
  28 def choose_cites(number, book=None, author=None):
  29     if book is not None:
  30         return book.choose_fragments(number) # todo: cites?
  31     elif author is not None:
  32         return Fragment.tagged.with_all([author]).order_by('?')[:number]
  33
  34
  35 @register.simple_tag(takes_context=True)
  36 def book_shelf_tags(context, book_id):
  37     request = context['request']
  38     if not request.user.is_authenticated:
  39         return ''
  40     book = Book.objects.get(pk=book_id)
  41     lks = likes(request.user, book, request)
  42
  43     def get_value():
  44         if not lks:
  45             return ''
  46         tags = book.tags.filter(category='set', user=request.user).exclude(name='')
  47         if not tags:
  48             return ''
  49         ctx = {'tags': tags}
  50         return template.loader.render_to_string('social/shelf_tags.html', ctx)
  51     return lazy(get_value, str)()
  52
  53
  54 @register.inclusion_tag('social/carousel.html', takes_context=True)
  55 def carousel(context, placement):
  56     banners = Carousel.get(placement).carouselitem_set.all()#first().get_banner()
  57     return {
  58         'banners': [b.get_banner() for b in banners],
  59     }
  60
  61
  62 @register.inclusion_tag('social/embed_video.html')
  63 def embed_video(url):
  64     m = re.match(r'https://www.youtube.com/watch\?v=([^&;]+)', url)
  65     return {
  66         'youtube_id': m.group(1) if m else None,
  67     }