X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/6e75a30246598a40cabb8ca1498c3a4f6f146c91..013ec8a416d4ff1ea3e1f2fa9ee2935623197ecd:/apps/catalogue/management/commands/pack.py

diff --git a/apps/catalogue/management/commands/pack.py b/apps/catalogue/management/commands/pack.py
index c75f092a9..ba06341b8 100755
--- a/apps/catalogue/management/commands/pack.py
+++ b/apps/catalogue/management/commands/pack.py
@@ -2,9 +2,6 @@
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
-import re
-import sys
-from cPickle import load, dump
 from optparse import make_option
 
 from django.core.management.base import BaseCommand
@@ -23,8 +20,8 @@ class Command(BaseCommand):
         make_option('-e', '--exclude', dest='exclude', metavar='SLUG,...',
             help='Exclude specific books by slug')
     )
-    help = 'Prepare data for Lesmianator.'
-    args = '[%s] output_path.zip' % '|'.join(ftypes)
+    help = 'Prepare ZIP package with files of given type.'
+    args = '[%s] output_path.zip' % '|'.join(Book.formats)
 
     def handle(self, ftype, path, **options):
         self.style = color_style()
@@ -33,7 +30,7 @@ class Command(BaseCommand):
         include = options.get('include')
         exclude = options.get('exclude')
 
-        if ftype in Book.file_types:
+        if ftype in Book.formats:
             field = "%s_file" % ftype
         else:
             print self.style.ERROR('Unknown file type.')