X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/9771b4edcbb556927312f624de6263395f82e0ba..5d3f4483d48655682297aa3c945c997633ea45d2:/apps/catalogue/templatetags/catalogue_tags.py
diff --git a/apps/catalogue/templatetags/catalogue_tags.py b/apps/catalogue/templatetags/catalogue_tags.py
index df938a624..e2aac35c7 100644
--- a/apps/catalogue/templatetags/catalogue_tags.py
+++ b/apps/catalogue/templatetags/catalogue_tags.py
@@ -2,21 +2,19 @@
# 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 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.forms import SearchForm
+from catalogue import forms
from catalogue.utils import split_tags
-
+from catalogue.models import Book, Fragment, Tag
register = template.Library()
@@ -48,6 +46,15 @@ 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):
title = []
@@ -260,17 +267,14 @@ 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/inline_tag_list.html')
def inline_tag_list(tags, choices=None):
- if choices is None:
- choices = []
- if len(tags) == 1:
- one_tag = tags[0]
- return locals()
+ return tag_list(tags, choices)
@register.inclusion_tag('catalogue/book_info.html')
@@ -278,21 +282,102 @@ def book_info(book):
return locals()
-@register.inclusion_tag('catalogue/book_wide.html')
-def book_wide(book):
- tags = book.tags.filter(category__in=('author', 'kind', 'genre', 'epoch'))
- tags = split_tags(tags)
+@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()
+ hide_about = extra_info.get('about', '').startswith('http://wiki.wolnepodreczniki.pl')
+
+ return {
+ 'book': book,
+ 'main_link': reverse('book_text', args=[book.slug]),
+ 'related': book.related_info(),
+ 'extra_info': book.get_extra_info_value(),
+ 'hide_about': hide_about,
+ 'themes': book_themes,
+ 'custom_pdf_form': forms.CustomPDFForm(),
+ 'request': context.get('request'),
+ }
+
+
+@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'),
+ }
+
+
+@register.inclusion_tag('catalogue/book_mini_box.html')
+def book_mini(book):
+ return {
+ 'book': book,
+ 'related': book.related_info(),
+ }
+
+
+@register.inclusion_tag('catalogue/work-list.html', takes_context=True)
+def work_list(context, object_list):
+ request = context.get('request')
+ if object_list:
+ object_type = type(object_list[0]).__name__
+ return locals()
- formats = {}
- # files generated during publication
- for ebook_format in book.ebook_formats:
- if book.has_media(ebook_format):
- formats[ebook_format] = book.get_media(ebook_format)
- extra_info = book.get_extra_info_value()
+@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:
+ related += list(Book.objects.exclude(
+ pk__in=[b.pk for b in related] + [book.pk]
+ ).order_by('?')[:random])
+ return {
+ 'books': related,
+ }
+
+
+@register.inclusion_tag('catalogue/menu.html')
+def catalogue_menu():
+ tags = Tag.objects.filter(
+ category__in=('author', 'epoch', 'genre', 'kind', 'theme')
+ ).exclude(book_count=0)
+ return split_tags(tags)
+
- has_media = {}
- for media_format in ['mp3', 'ogg']:
- has_media[media_format] = book.has_media(media_format)
- return locals()
+@register.simple_tag
+def tag_url(category, slug):
+ return reverse('catalogue.views.tagged_object_list', args=[
+ '/'.join((Tag.categories_dict[category], slug))
+ ])