-# -*- coding: utf-8 -*-
# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
#
-from random import randint
-from django.db.models import Q
+import re
from django import template
from django.utils.functional import lazy
from django.utils.cache import add_never_cache_headers
-from catalogue.models import Book, Tag
-from ssify import ssi_variable
-from ssify.utils import ssi_vary_on_cookie
-from social.models import Cite
-from social.utils import likes, cites_for_tags
+from catalogue.models import Book
+from social.utils import likes, get_or_choose_cite
+from ..models import Carousel, Cite
register = template.Library()
-@ssi_variable(register, patch_response=[ssi_vary_on_cookie])
-def likes_book(request, book_id):
- return likes(request.user, Book.objects.get(pk=book_id), request)
+@register.simple_tag(takes_context=True)
+def likes_book(context, book):
+ request = context['request']
+ return likes(request.user, book, request)
-def choose_cite(book_id=None, tag_ids=None):
- """Choose a cite for main page, for book or for set of tags."""
- if book_id is not None:
- cites = Cite.objects.filter(Q(book=book_id) | Q(book__ancestor=book_id))
- elif tag_ids is not None:
- tags = Tag.objects.filter(pk__in=tag_ids)
- cites = cites_for_tags(tags)
- else:
- cites = Cite.objects.all()
- stickies = cites.filter(sticky=True)
- count = stickies.count()
- if count:
- cite = stickies[randint(0, count - 1)]
- else:
- count = cites.count()
- if count:
- cite = cites[randint(0, count - 1)]
- else:
- cite = None
- return cite
-
-
-@ssi_variable(register, name='choose_cite', patch_response=[add_never_cache_headers])
-def choose_cite_tag(request, book_id=None, tag_ids=None):
- try:
- assert request.user.is_staff
- assert 'choose_cite' in request.GET
- cite = Cite.objects.get(pk=request.GET['choose_cite'])
- except (AssertionError, Cite.DoesNotExist):
- cite = choose_cite(book_id, tag_ids)
- return cite.pk if cite is not None else None
+@register.simple_tag(takes_context=True)
+def choose_cite(context, book_id=None, tag_ids=None):
+ request = context['request']
+ return get_or_choose_cite(request, book_id, tag_ids)
-@register.inclusion_tag('social/cite_promo.html')
-def render_cite(cite):
- return {
- 'cite': cite,
- }
-
-
-@ssi_variable(register, patch_response=[ssi_vary_on_cookie])
-def book_shelf_tags(request, book_id):
- if not request.user.is_authenticated():
- return None
+@register.simple_tag(takes_context=True)
+def book_shelf_tags(context, book_id):
+ request = context['request']
+ if not request.user.is_authenticated:
+ return ''
book = Book.objects.get(pk=book_id)
lks = likes(request.user, book, request)
return ''
ctx = {'tags': tags}
return template.loader.render_to_string('social/shelf_tags.html', ctx)
- return lazy(get_value, unicode)()
+ return lazy(get_value, str)()
+
+
+@register.inclusion_tag('social/carousel.html', takes_context=True)
+def carousel(context, placement):
+ carousel = Carousel.get(placement)
+ banners = [
+ item.get_banner()
+ for item in carousel.carouselitem_set.all().select_related('banner')
+ ]
+
+ request = context['request']
+ if 'banner' in request.GET:
+ try:
+ banner_id = int(request.GET['banner'])
+ except (TypeError, ValueError):
+ pass
+ else:
+ try:
+ index = [b.pk for b in banners].index(banner_id)
+ except ValueError:
+ if request.user.is_staff:
+ # Staff is allowed to preview any banner.
+ try:
+ banners.insert(0, Cite.objects.get(pk=banner_id))
+ except Cite.DoesNotExist:
+ pass
+ else:
+ # Put selected banner to front.
+ banners = [banners[index]] + banners[:index] + banners[index+1:]
+
+ return {
+ 'carousel': carousel,
+ 'banners': banners,
+ }
+
+
+@register.inclusion_tag('social/embed_video.html')
+def embed_video(url):
+ m = re.match(r'https://www.youtube.com/watch\?v=([^&;]+)', url)
+ return {
+ 'youtube_id': m.group(1) if m else None,
+ }
fnp / wolnelektury.git / blobdiff