X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/79ac7d1116560971a70246601957995789536f9f..HEAD:/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
new mode 100644
index 484b52ecb..a4b0f3ed2
--- a/src/social/templatetags/social_tags.py
+++ b/src/social/templatetags/social_tags.py
@@ -1,70 +1,42 @@
-# -*- 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.
+# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. 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, Fragment
+from social.utils import likes, get_or_choose_cite, choose_cite as cs
+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
+@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)
 
 
-@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
+def choose_cites(number, book=None, author=None):
+    if book is not None:
+        return book.choose_fragments(number) # todo: cites?
+    elif author is not None:
+        return Fragment.tagged.with_all([author]).order_by('?')[:number]
 
 
-@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)
 
@@ -76,4 +48,20 @@ 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, placement):
+    banners = Carousel.get(placement).carouselitem_set.all()#first().get_banner()
+    return {
+        'banners': [b.get_banner() for b in 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,
+    }