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

diff --git a/src/social/templatetags/social_tags.py b/src/social/templatetags/social_tags.py
index 0135792ce..88d6771db 100644
--- a/src/social/templatetags/social_tags.py
+++ b/src/social/templatetags/social_tags.py
@@ -5,7 +5,7 @@ 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 catalogue.models import Book, Fragment
 from social.utils import likes, get_or_choose_cite, choose_cite as cs
 from ..models import Carousel, Cite
 
@@ -25,8 +25,11 @@ def choose_cite(context, book_id=None, tag_ids=None):
 
 
 @register.simple_tag
-def choose_cites(book, number):
-    return  book.choose_fragments(number) # todo: cites?
+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)