X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/bb551d54ea65621cae28c6acbfd91b22dc441dae..577bb2b84d936d65bf072ef0a6b898db9d6e77ab:/apps/wiki/helpers.py?ds=sidebyside

diff --git a/apps/wiki/helpers.py b/apps/wiki/helpers.py
index 253f1291..3e0267ed 100644
--- a/apps/wiki/helpers.py
+++ b/apps/wiki/helpers.py
@@ -152,7 +152,7 @@ class ChunksList(object):
             'book', 'stage__name',
             'user')
 
-        self.book_ids = [x['book_id'] for x in chunk_qs.values('book_id')]
+        self.book_qs = chunk_qs.values('book_id')
 
     def __getitem__(self, key):
         if isinstance(key, slice):
@@ -163,10 +163,10 @@ class ChunksList(object):
             raise TypeError('Unsupported list index. Must be a slice or an int.')
 
     def __len__(self):
-        return len(self.book_ids)
+        return self.book_qs.count()
 
     def get_slice(self, slice_):
-        book_ids = self.book_ids[slice_]
+        book_ids = [x['book_id'] for x in self.book_qs[slice_]]
         chunk_qs = self.chunk_qs.filter(book__in=book_ids)
 
         chunks_list = []