X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/57a20265fc4e83cdaa26d8c05af2e4d9a77afea0..c6db46d42f0a6b9dbd5abb6ce2be58fe306752b6:/apps/catalogue/templatetags/catalogue_tags.py?ds=inline diff --git a/apps/catalogue/templatetags/catalogue_tags.py b/apps/catalogue/templatetags/catalogue_tags.py index ad60962c5..92827c10d 100644 --- a/apps/catalogue/templatetags/catalogue_tags.py +++ b/apps/catalogue/templatetags/catalogue_tags.py @@ -4,6 +4,7 @@ # import datetime import feedparser +from random import randint from django.conf import settings from django import template @@ -388,9 +389,14 @@ def related_books(book, limit=6, random=1): ignore_by_tag=book.book_tag())[:limit-random] cache.set(cache_key, related, 1800) if random: - related += list(Book.objects.exclude( - pk__in=[b.pk for b in related] + [book.pk] - ).order_by('?')[:random]) + random_books = Book.objects.exclude( + pk__in=[b.pk for b in related] + [book.pk]) + if random == 1: + count = random_books.count() + if count: + related.append(random_books[randint(0, count - 1)]) + else: + related += list(random_books.order_by('?')[:random]) return { 'books': related, } @@ -407,9 +413,7 @@ def catalogue_menu(): @register.simple_tag def tag_url(category, slug): - return reverse('catalogue.views.tagged_object_list', args=[ - '/'.join((Tag.categories_dict[category], slug)) - ]) + return Tag.create_url(category, slug) @register.simple_tag