X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/75957f735219259d3b4bc361f80ccd3d7b92a0e9..592c9401b18118e01de250d44878b9289b138f9c:/src/social/templatetags/social_tags.py

diff --git a/src/social/templatetags/social_tags.py b/src/social/templatetags/social_tags.py
index 298782c52..88d6771db 100644
--- a/src/social/templatetags/social_tags.py
+++ b/src/social/templatetags/social_tags.py
@@ -5,8 +5,8 @@ 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
-from social.utils import likes, get_or_choose_cite
+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()
@@ -24,10 +24,18 @@ def choose_cite(context, book_id=None, tag_ids=None):
     return get_or_choose_cite(request, book_id, tag_ids)
 
 
+@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.simple_tag(takes_context=True)
 def book_shelf_tags(context, book_id):
     request = context['request']
-    if not request.user.is_authenticated():
+    if not request.user.is_authenticated:
         return ''
     book = Book.objects.get(pk=book_id)
     lks = likes(request.user, book, request)
@@ -44,13 +52,8 @@ def book_shelf_tags(context, book_id):
 
 
 @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 = Carousel.objects.create(slug=slug)
+def carousel(context, placement):
+    carousel = Carousel.get(placement)
     banners = [
             item.get_banner()
             for item in carousel.carouselitem_set.all().select_related('banner')