fixes #2581: Hide themes from menu if no themes.

[wolnelektury.git] / apps / catalogue / models / collection.py

diff --git a/apps/catalogue/models/collection.py b/apps/catalogue/models/collection.py
index 1c9027c..8f67f88 100644 (file)
--- a/apps/catalogue/models/collection.py
+++ b/apps/catalogue/models/collection.py
@@ -23,3 +23,14 @@ class Collection(models.Model):
 
     def __unicode__(self):
         return self.title
 
     def __unicode__(self):
         return self.title

+
+    @models.permalink
+    def get_absolute_url(self):
+        return ("collection", [self.slug])
+
+    def get_query(self):
+        slugs = self.book_slugs.split()
+        # allow URIs
+        slugs = [slug.rstrip('/').rsplit('/', 1)[-1] if '/' in slug else slug
+                    for slug in slugs]
+        return models.Q(slug__in=slugs)