X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/97d7b46edcc444cacafc2e4824618bfc322339c0..1e9113af163a3d002b0b9c3e8c339f22af5b848a:/apps/catalogue/templatetags/catalogue_tags.py
diff --git a/apps/catalogue/templatetags/catalogue_tags.py b/apps/catalogue/templatetags/catalogue_tags.py
index 94753483c..e80a5ffe6 100644
--- a/apps/catalogue/templatetags/catalogue_tags.py
+++ b/apps/catalogue/templatetags/catalogue_tags.py
@@ -2,17 +2,21 @@
# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
#
-import feedparser
import datetime
+import feedparser
+from random import randint
+from django.conf import settings
from django import template
-from django.template import Node, Variable
-from django.utils.encoding import smart_str
+from django.template import Node, Variable, Template, Context
+from django.core.cache import cache
from django.core.urlresolvers import reverse
from django.contrib.auth.forms import UserCreationForm, AuthenticationForm
-from django.db.models import Q
-from django.conf import settings
+from django.utils.translation import ugettext as _
+from catalogue.utils import split_tags, related_tag_name as _related_tag_name
+from catalogue.models import Book, BookMedia, Fragment, Tag
+from catalogue.constants import LICENSES, LANGUAGES_3TO2
register = template.Library()
@@ -44,46 +48,25 @@ def capfirst(text):
return ''
+@register.simple_tag
+def html_title_from_tags(tags):
+ if len(tags) < 2:
+ return title_from_tags(tags)
+ template = Template("{{ category }}: {{ tag.name }}")
+ return capfirst(",
".join(
+ template.render(Context({'tag': tag, 'category': _(tag.category)})) for tag in tags))
+
def simple_title(tags):
- mapping = {
- 'author': u'Autor',
- 'theme': u'motyw',
- 'epoch': u'epoka',
- 'genre': u'gatunek',
- 'kind': u'rodzaj',
- 'set': u'póÅka',
- }
-
title = []
for tag in tags:
- title.append("%s: %s" % (mapping[tag.category], tag.name))
+ title.append("%s: %s" % (_(tag.category), tag.name))
return capfirst(', '.join(title))
-def book_stub_title(book):
- return ', '.join((book.author, book.title))
-
-
@register.simple_tag
def book_title(book, html_links=False):
- try:
- names = list(book.tags.filter(category='author'))
- except AttributeError:
- return book_stub_title(book)
-
- books = []
- while book:
- books.append(book)
- book = book.parent
- names.extend(reversed(books))
-
- if html_links:
- names = ['%s' % (tag.get_absolute_url(), tag.name) for tag in names]
- else:
- names = [tag.name for tag in names]
-
- return ', '.join(names)
+ return book.pretty_title(html_links)
@register.simple_tag
@@ -155,6 +138,85 @@ def title_from_tags(tags):
return capfirst(title)
+@register.simple_tag
+def book_tree(book_list, books_by_parent):
+ text = "".join("%s%s" % (
+ book.get_absolute_url(), book.title, book_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 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("""
+ %(depth)dem%(title)s
+ %(audiences)s
+ %(audiobook)s
+
+ %(children)s
+ """ % {
+ "depth": depth,
+ "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 = []
+ if 'editors' in extra_info:
+ editors += extra_info['editors']
+ if 'technical_editors' in extra_info:
+ editors += extra_info['technical_editors']
+ # support for extra_info-s from librarian<1.2
+ if 'editor' in extra_info:
+ editors.append(extra_info['editor'])
+ if 'technical_editor' in extra_info:
+ editors.append(extra_info['technical_editor'])
+ return ', '.join(
+ ' '.join(p.strip() for p in person.rsplit(',', 1)[::-1])
+ for person in sorted(set(editors)))
+
+
@register.simple_tag
def user_creation_form():
return RegistrationForm(prefix='registration').as_ul()
@@ -165,23 +227,9 @@ def authentication_form():
return LoginForm(prefix='login').as_ul()
-@register.inclusion_tag('catalogue/breadcrumbs.html')
-def breadcrumbs(tags, search_form=True):
- from catalogue.forms import SearchForm
- context = {'tag_list': tags}
- try:
- max_tag_list = settings.MAX_TAG_LIST
- except AttributeError:
- max_tag_list = -1
- if search_form and (max_tag_list == -1 or len(tags) < max_tag_list):
- context['search_form'] = SearchForm(tags=tags)
- return context
-
-
@register.tag
def catalogue_url(parser, token):
bits = token.split_contents()
- tag_name = bits[0]
tags_to_add = []
tags_to_remove = []
@@ -253,24 +301,181 @@ def latest_blog_posts(feed_url, posts_to_show=5):
def tag_list(tags, choices=None):
if choices is None:
choices = []
- if len(tags) == 1:
+ if len(tags) == 1 and tags[0].category not in [t.category for t in choices]:
one_tag = tags[0]
return locals()
-@register.inclusion_tag('catalogue/folded_tag_list.html')
-def folded_tag_list(tags, choices=None):
- tags = [tag for tag in tags if tag.count]
- if choices is None:
- choices = []
- some_tags_hidden = False
- tag_count = len(tags)
+@register.inclusion_tag('catalogue/inline_tag_list.html')
+def inline_tag_list(tags, choices=None):
+ return tag_list(tags, choices)
- if tag_count == 1:
- one_tag = tags[0]
- else:
- shown_tags = [tag for tag in tags if tag.main_page]
- if tag_count > len(shown_tags):
- some_tags_hidden = True
+
+@register.inclusion_tag('catalogue/collection_list.html')
+def collection_list(collections):
+ return locals()
+
+
+@register.inclusion_tag('catalogue/book_info.html')
+def book_info(book):
return locals()
+
+@register.inclusion_tag('catalogue/book_wide.html', takes_context=True)
+def book_wide(context, book):
+ 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]) if book.html_file else None,
+ 'related': book.related_info(),
+ 'extra_info': extra_info,
+ 'hide_about': hide_about,
+ 'themes': book_themes,
+ 'request': context.get('request'),
+ 'show_lang': book.language_code() != settings.LANGUAGE_CODE,
+ }
+
+
+@register.inclusion_tag('catalogue/book_short.html', takes_context=True)
+def book_short(context, book):
+ return {
+ 'book': book,
+ 'main_link': book.get_absolute_url(),
+ 'related': book.related_info(),
+ 'request': context.get('request'),
+ 'show_lang': book.language_code() != settings.LANGUAGE_CODE,
+ }
+
+
+@register.inclusion_tag('catalogue/book_mini_box.html')
+def book_mini(book, with_link=True):
+ author_str = ", ".join(related_tag_name(tag)
+ for tag in book.related_info()['tags'].get('author', ()))
+ return {
+ 'book': book,
+ 'author_str': author_str,
+ 'with_link': with_link,
+ 'show_lang': book.language_code() != settings.LANGUAGE_CODE,
+ }
+
+
+@register.inclusion_tag('catalogue/work-list.html', takes_context=True)
+def work_list(context, object_list):
+ request = context.get('request')
+ for obj in object_list:
+ obj.object_type = type(object_list[0]).__name__
+
+ return locals()
+
+
+@register.inclusion_tag('catalogue/fragment_promo.html')
+def fragment_promo(arg=None):
+ if arg is None:
+ fragments = Fragment.objects.all().order_by('?')
+ fragment = fragments[0] if fragments.exists() else None
+ elif isinstance(arg, Book):
+ fragment = arg.choose_fragment()
+ else:
+ fragments = Fragment.tagged.with_all(arg).order_by('?')
+ fragment = fragments[0] if fragments.exists() else None
+
+ return {
+ 'fragment': fragment,
+ }
+
+
+@register.inclusion_tag('catalogue/related_books.html')
+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:
+ related = list(Book.objects.filter(
+ common_slug=book.common_slug).exclude(pk=book.pk)[:limit])
+ limit -= len(related)
+ if limit > random:
+ related += Book.tagged.related_to(book,
+ Book.objects.exclude(common_slug=book.common_slug),
+ ignore_by_tag=book.book_tag())[:limit-random]
+ cache.set(cache_key, related, 1800)
+ if 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,
+ }
+
+
+@register.inclusion_tag('catalogue/menu.html')
+def catalogue_menu():
+ return {'categories': [
+ ('author', _('Authors'), 'autorzy'),
+ ('genre', _('Genres'), 'gatunki'),
+ ('kind', _('Kinds'), 'rodzaje'),
+ ('epoch', _('Epochs'), 'epoki'),
+ ('theme', _('Themes'), 'autorzy'),
+ ]}
+
+
+@register.simple_tag
+def tag_url(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'],
+ }
+
+
+@register.simple_tag
+def related_tag_name(tag, lang=None):
+ return _related_tag_name(tag, lang)
+
+
+@register.simple_tag
+def class_name(obj):
+ return obj.__class__.__name__
+