X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/a42b731727d0c4b8e372d001c5a1be776067eae8..e34fd1a772c0b534073adba2377502eced4d36ae:/apps/catalogue/models/collection.py

diff --git a/apps/catalogue/models/collection.py b/apps/catalogue/models/collection.py
index daee48461..8f67f88e0 100644
--- a/apps/catalogue/models/collection.py
+++ b/apps/catalogue/models/collection.py
@@ -27,3 +27,10 @@ class Collection(models.Model):
     @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)