Added inclusion tag for book.

[wolnelektury.git] / catalogue / templatetags / catalogue_tags.py

diff --git a/catalogue/templatetags/catalogue_tags.py b/catalogue/templatetags/catalogue_tags.py
index 3c75801..d3a25ad 100644 (file)
--- a/catalogue/templatetags/catalogue_tags.py
+++ b/catalogue/templatetags/catalogue_tags.py
@@ -4,6 +4,7 @@ from django.template import Node, Variable
 from django.utils.encoding import smart_str
 from django.core.urlresolvers import reverse
 from django.contrib.auth.forms import UserCreationForm, AuthenticationForm
+from django.db.models import Q
 
 
 register = template.Library()
@@ -116,6 +117,22 @@ def breadcrumbs(tags, search_form=True):
     return context
 
 
+@register.inclusion_tag('catalogue/_book.html')
+def book(book):
+    tags = book.tags.filter(~Q(category__in=('set', 'theme')))
+    tags = [u'<a href="%s">%s</a>' % (tag.get_absolute_url(), tag.name) for tag in tags]
+    
+    formats = []
+    if book.html_file:
+        formats.append(u'<a href="%s">Czytaj online</a>' % book.html_file.url)
+    if book.pdf_file:
+        formats.append(u'<a href="%s">Plik PDF</a>' % book.pdf_file.url)
+    if book.odt_file:
+        formats.append(u'<a href="%s">Plik ODT</a>' % book.odt_file.url)
+        
+    return {'book': book, 'tags': tags, 'formats': formats}
+
+
 @register.tag
 def catalogue_url(parser, token):
     bits = token.split_contents()