X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/3748094efcd5ed3b89408ece388a6e94ea42990c..c6db46d42f0a6b9dbd5abb6ce2be58fe306752b6:/apps/catalogue/templatetags/catalogue_tags.py diff --git a/apps/catalogue/templatetags/catalogue_tags.py b/apps/catalogue/templatetags/catalogue_tags.py index e8bc13899..92827c10d 100644 --- a/apps/catalogue/templatetags/catalogue_tags.py +++ b/apps/catalogue/templatetags/catalogue_tags.py @@ -4,7 +4,9 @@ # import datetime import feedparser +from random import randint +from django.conf import settings from django import template from django.template import Node, Variable, Template, Context from django.core.cache import cache @@ -12,9 +14,9 @@ from django.core.urlresolvers import reverse from django.contrib.auth.forms import UserCreationForm, AuthenticationForm from django.utils.translation import ugettext as _ -from catalogue import forms from catalogue.utils import split_tags -from catalogue.models import Book, Fragment, Tag +from catalogue.models import Book, BookMedia, Fragment, Tag +from catalogue.constants import LICENSES register = template.Library() @@ -55,7 +57,6 @@ def html_title_from_tags(tags): template.render(Context({'tag': tag, 'category': _(tag.category)})) for tag in tags)) - def simple_title(tags): title = [] for tag in tags: @@ -148,6 +149,17 @@ def book_tree(book_list, books_by_parent): else: return '' +@register.simple_tag +def audiobook_tree(book_list, books_by_parent): + text = "".join("
  • %s%s
  • " % ( + reverse("book_player", args=[book.slug]), book.title, audiobook_tree(books_by_parent.get(book, ()), books_by_parent) + ) for book in book_list) + + if text: + return "
      %s
    " % text + else: + return '' + @register.simple_tag def book_tree_texml(book_list, books_by_parent, depth=1): return "".join(""" @@ -158,13 +170,36 @@ def book_tree_texml(book_list, books_by_parent, depth=1): %(children)s """ % { "depth": depth, - "title": book.title, + "title": book.title, "audiences": ", ".join(book.audiences_pl()), "audiobook": "audiobook" if book.has_media('mp3') else "", "children": book_tree_texml(books_by_parent.get(book.id, ()), books_by_parent, depth + 1) } for book in book_list) +@register.simple_tag +def book_tree_csv(author, book_list, books_by_parent, depth=1, max_depth=3, delimeter="\t"): + def quote_if_necessary(s): + try: + s.index(delimeter) + s.replace('"', '\\"') + return '"%s"' % s + except ValueError: + return s + + return "".join("""%(author)s%(d)s%(preindent)s%(title)s%(d)s%(postindent)s%(audiences)s%(d)s%(audiobook)s +%(children)s""" % { + "d": delimeter, + "preindent": delimeter * (depth - 1), + "postindent": delimeter * (max_depth - depth), + "depth": depth, + "author": quote_if_necessary(author.name), + "title": quote_if_necessary(book.title), + "audiences": ", ".join(book.audiences_pl()), + "audiobook": "audiobook" if book.has_media('mp3') else "", + "children": book_tree_csv(author, books_by_parent.get(book.id, ()), books_by_parent, depth + 1) + } for book in book_list) + @register.simple_tag def all_editors(extra_info): editors = [] @@ -195,7 +230,6 @@ def authentication_form(): @register.tag def catalogue_url(parser, token): bits = token.split_contents() - tag_name = bits[0] tags_to_add = [] tags_to_remove = [] @@ -284,21 +318,17 @@ def book_info(book): @register.inclusion_tag('catalogue/book_wide.html', takes_context=True) def book_wide(context, book): - theme_counter = book.theme_counter - book_themes = Tag.objects.filter(pk__in=theme_counter.keys()) - for tag in book_themes: - tag.count = theme_counter[tag.pk] - extra_info = book.get_extra_info_value() + book_themes = book.related_themes() + extra_info = book.extra_info hide_about = extra_info.get('about', '').startswith('http://wiki.wolnepodreczniki.pl') return { 'book': book, - 'main_link': reverse('book_text', args=[book.slug]), + 'main_link': reverse('book_text', args=[book.slug]) if book.html_file else None, 'related': book.related_info(), - 'extra_info': book.get_extra_info_value(), + 'extra_info': extra_info, 'hide_about': hide_about, 'themes': book_themes, - 'custom_pdf_form': forms.CustomPDFForm(), 'request': context.get('request'), } @@ -350,7 +380,6 @@ def related_books(book, limit=6, random=1): cache_key = "catalogue.related_books.%d.%d" % (book.id, limit - random) related = cache.get(cache_key) if related is None: - print 'not in cache' related = list(Book.objects.filter( common_slug=book.common_slug).exclude(pk=book.pk)[:limit]) limit -= len(related) @@ -360,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, } @@ -379,6 +413,44 @@ 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 +def download_audio(book, daisy=True): + related = book.related_info() + links = [] + if related['media'].get('mp3'): + links.append("%s" % + (reverse('download_zip_mp3', args=[book.slug]), + BookMedia.formats['mp3'].name)) + if related['media'].get('ogg'): + links.append("%s" % + (reverse('download_zip_ogg', args=[book.slug]), + BookMedia.formats['ogg'].name)) + if daisy and related['media'].get('daisy'): + for dsy in book.get_media('daisy'): + links.append("%s" % + (dsy.file.url, BookMedia.formats['daisy'].name)) + return ", ".join(links) + + +@register.inclusion_tag("catalogue/snippets/custom_pdf_link_li.html") +def custom_pdf_link_li(book): + return { + 'book': book, + 'NO_CUSTOM_PDF': settings.NO_CUSTOM_PDF, + } + + +@register.inclusion_tag("catalogue/snippets/license_icon.html") +def license_icon(license_url): + """Creates a license icon, if the license_url is known.""" + known = LICENSES.get(license_url) + if known is None: + return {} + return { + "license_url": license_url, + "icon": "img/licenses/%s.png" % known['icon'], + "license_description": known['description'], + }