X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/d91551345b68c2bc7d96f2098691fab28276d6b8..0c1644b480a81a3b1d55efcd4fc328f7f7ed77a6:/apps/catalogue/management/commands/pack.py diff --git a/apps/catalogue/management/commands/pack.py b/apps/catalogue/management/commands/pack.py index 80f612ad3..6ecf32d59 100755 --- a/apps/catalogue/management/commands/pack.py +++ b/apps/catalogue/management/commands/pack.py @@ -23,9 +23,8 @@ class Command(BaseCommand): make_option('-e', '--exclude', dest='exclude', metavar='SLUG,...', help='Exclude specific books by slug') ) - help = 'Prepare data for Lesmianator.' - ftypes = ['xml', 'txt', 'html', 'epub', 'pdf'] - 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() @@ -34,7 +33,7 @@ class Command(BaseCommand): include = options.get('include') exclude = options.get('exclude') - if ftype in self.ftypes: + if ftype in Book.formats: field = "%s_file" % ftype else: print self.style.ERROR('Unknown file type.') @@ -53,7 +52,7 @@ class Command(BaseCommand): if exclude: books = [book for book in books if book.slug not in exclude.split(',')] - archive = zipfile.ZipFile(path, 'w') + archive = zipfile.ZipFile(path, 'w', zipfile.ZIP_DEFLATED) processed = skipped = 0 for book in books: