Style fix

[wolnelektury.git] / src / catalogue / management / commands / pack.py

diff --git a/src/catalogue/management/commands/pack.py b/src/catalogue/management/commands/pack.py
index 181c2e6..22b36e9 100644 (file)
--- a/src/catalogue/management/commands/pack.py
+++ b/src/catalogue/management/commands/pack.py
@@ -20,6 +20,8 @@ class Command(BaseCommand):
         parser.add_argument(
                 '-e', '--exclude', dest='exclude', metavar='SLUG,...',
                 help='Exclude specific books by slug')
         parser.add_argument(
                 '-e', '--exclude', dest='exclude', metavar='SLUG,...',
                 help='Exclude specific books by slug')

+        parser.add_argument(
+                '--top-level', dest='top_level', action='store_true')

         parser.add_argument('ftype', metavar='|'.join(Book.formats))
         parser.add_argument('path', metavar='output_path.zip')
 
         parser.add_argument('ftype', metavar='|'.join(Book.formats))
         parser.add_argument('path', metavar='output_path.zip')
 


@@ -31,6 +33,7 @@ class Command(BaseCommand):
         tags = options.get('tags')
         include = options.get('include')
         exclude = options.get('exclude')
         tags = options.get('tags')
         include = options.get('include')
         exclude = options.get('exclude')

+        top_level = options.get('top_level')

 
         if ftype in Book.formats:
             field = "%s_file" % ftype
 
         if ftype in Book.formats:
             field = "%s_file" % ftype


@@ -46,7 +49,10 @@ class Command(BaseCommand):
         if tags:
             books += list(Book.tagged.with_all(Tag.objects.filter(slug__in=tags.split(','))).only('slug', field))
         elif not include:
         if tags:
             books += list(Book.tagged.with_all(Tag.objects.filter(slug__in=tags.split(','))).only('slug', field))
         elif not include:

-            books = list(Book.objects.all().only('slug', field))
+            books = Book.objects.all()
+            if top_level:
+                books = books.filter(parent=None)
+            books = list(books.only('slug', field))

 
         if exclude:
             books = [book for book in books if book.slug not in exclude.split(',')]
 
         if exclude:
             books = [book for book in books if book.slug not in exclude.split(',')]