193a1ce3a03d67eaaf001798474817cb7a11ee19
[wolnelektury.git] / 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, 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     carousel = Carousel.get(placement)
57     banners = [
58             item.get_banner()
59             for item in carousel.carouselitem_set.all().select_related('banner')
60             ]
61
62     request = context['request']
63     if 'banner' in request.GET:
64         try:
65             banner_id = int(request.GET['banner'])
66         except (TypeError, ValueError):
67             pass
68         else:
69             try:
70                 index = [b.pk for b in banners].index(banner_id)
71             except ValueError:
72                 if request.user.is_staff:
73                     # Staff is allowed to preview any banner.
74                     try:
75                         banners.insert(0, Cite.objects.get(pk=banner_id))
76                     except Cite.DoesNotExist:
77                         pass
78             else:
79                 # Put selected banner to front.
80                 banners = [banners[index]] + banners[:index] + banners[index+1:]
81
82     return {
83         'carousel': carousel,
84         'banners': banners,
85     }
86
87
88 @register.inclusion_tag('social/carousel_2022.html', takes_context=True)
89 def carousel_2022(context, placement):
90     banner = Carousel.get(placement).carouselitem_set.first().get_banner()
91     return {
92         'banner': banner,
93     }
94
95
96 @register.inclusion_tag('social/embed_video.html')
97 def embed_video(url):
98     m = re.match(r'https://www.youtube.com/watch\?v=([^&;]+)', url)
99     return {
100         'youtube_id': m.group(1) if m else None,
101     }