Cite/banner/carousel change.

[wolnelektury.git] / src / social / templatetags / social_tags.py

diff --git a/src/social/templatetags/social_tags.py b/src/social/templatetags/social_tags.py
old mode 100755 (executable)
new mode 100644 (file)
index ce61985..a444d66
--- a/src/social/templatetags/social_tags.py
+++ b/src/social/templatetags/social_tags.py
@@ -1,7 +1,7 @@
-# -*- 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.
 #
+import re
 from django import template
 from django.utils.functional import lazy
 from django.utils.cache import add_never_cache_headers
@@ -9,6 +9,7 @@ from catalogue.models import Book
 from ssify import ssi_variable
 from ssify.utils import ssi_vary_on_cookie
 from social.utils import likes, get_or_choose_cite
+from ..models import Carousel, Cite
 
 register = template.Library()
 
@@ -46,4 +47,51 @@ def book_shelf_tags(request, book_id):
             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, slug):
+    # TODO: cache
+    try:
+        carousel = Carousel.objects.get(slug=slug)
+    except Carousel.DoesNotExist:
+        # TODO: add sanity check for install.
+        carousel = None
+    banners = [
+            item.get_banner()
+            for item in carousel.carouselitem_set.all()
+            ]
+
+    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,
+    }