From 9ba3ccd2ad2748ed4a1b67187b8e1ac273dc8ee8 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Marek=20St=C4=99pniowski?= <marek@stepniowski.com>
Date: Thu, 4 Sep 2008 00:49:59 +0200
Subject: [PATCH] Changed rendering of fragment lists.

---
 catalogue/views.py                        | 11 +++++++++--
 templates/catalogue/main_page.html        |  2 +-
 templates/catalogue/tagged_book_list.html |  2 +-
 3 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/catalogue/views.py b/catalogue/views.py
index 49f41965f..98d3f942f 100644
--- a/catalogue/views.py
+++ b/catalogue/views.py
@@ -75,6 +75,8 @@ def main_page(request):
     else:
         extra_where = 'NOT catalogue_tag.category = "set"'
     tags = models.Tag.objects.usage_for_model(models.Book, counts=True, extra={'where': [extra_where]})
+    fragment_tags = models.Tag.objects.usage_for_model(models.Fragment, counts=True,
+        extra={'where': ['catalogue_tag.category = "theme"']})
     categories = split_tags(tags)
     
     form = forms.SearchForm()
@@ -99,12 +101,17 @@ def tagged_book_list(request, tags=''):
         tags = models.Tag.get_tag_list(tags)
     except models.Tag.DoesNotExist:
         raise Http404
-        
+    
+    model = models.Book
+    theme_is_set = any(tag.category == 'theme' for tag in tags)
+    if theme_is_set:
+        model = models.Fragment
+    
     if request.user.is_authenticated():
         extra_where = '(NOT catalogue_tag.category = "set" OR catalogue_tag.user_id = %d)' % request.user.id
     else:
         extra_where = 'NOT catalogue_tag.category = "set"'
-    related_tags = models.Tag.objects.related_for_model(tags, models.Book, counts=True, extra={'where': [extra_where]})
+    related_tags = models.Tag.objects.related_for_model(tags, model, counts=True, extra={'where': [extra_where]})
     categories = split_tags(related_tags)
 
     return tagged_object_list(
diff --git a/templates/catalogue/main_page.html b/templates/catalogue/main_page.html
index db7c95de3..221595fbf 100644
--- a/templates/catalogue/main_page.html
+++ b/templates/catalogue/main_page.html
@@ -72,7 +72,7 @@
             {% if categories.theme %}
                 <h2>Motywy</h2>
                 <ul>
-                {% for theme in categories.theme %}
+                {% for theme in fragment_tags %}
                     <li><a href="{% catalogue_url theme %}">{{ theme }}&nbsp;({{ theme.count }})</a></li>
                 {% endfor %}
                 </ul>                
diff --git a/templates/catalogue/tagged_book_list.html b/templates/catalogue/tagged_book_list.html
index 329b4df6c..cca0e56d4 100644
--- a/templates/catalogue/tagged_book_list.html
+++ b/templates/catalogue/tagged_book_list.html
@@ -68,7 +68,7 @@
         {% endwith %}
         <ol>
         {% for book in object_list %}
-            <li>{% include "catalogue/_book.html" %}</li>
+            <li>{% book book %}</li>
         {% endfor %}
         </ol>
         {% paginate %}
-- 
2.20.1