Check field limits on picture import.
[wolnelektury.git] / apps / catalogue / management / commands / pack.py
1 # -*- coding: utf-8 -*-
2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from optparse import make_option
6
7 from django.core.management.base import BaseCommand
8 from django.core.management.color import color_style
9 import zipfile
10
11 from catalogue.models import Book, Tag
12
13
14 class Command(BaseCommand):
15     option_list = BaseCommand.option_list + (
16         make_option('-t', '--tags', dest='tags', metavar='SLUG,...',
17             help='Use only books tagged with this tags'),
18         make_option('-i', '--include', dest='include', metavar='SLUG,...',
19             help='Include specific books by slug'),
20         make_option('-e', '--exclude', dest='exclude', metavar='SLUG,...',
21             help='Exclude specific books by slug')
22     )
23     help = 'Prepare ZIP package with files of given type.'
24     args = '[%s] output_path.zip' % '|'.join(Book.formats)
25
26     def handle(self, ftype, path, **options):
27         self.style = color_style()
28         verbose = int(options.get('verbosity'))
29         tags = options.get('tags')
30         include = options.get('include')
31         exclude = options.get('exclude')
32
33         if ftype in Book.formats:
34             field = "%s_file" % ftype
35         else:
36             print self.style.ERROR('Unknown file type.')
37             return
38
39         books = []
40
41         if include:
42             books += list(Book.objects.filter(slug__in=include.split(',')).only('slug', field))
43
44         if tags:
45             books += list(Book.tagged.with_all(Tag.objects.filter(slug__in=tags.split(','))).only('slug', field))
46         elif not include:
47             books = list(Book.objects.all().only('slug', field))
48
49         if exclude:
50             books = [book for book in books if book.slug not in exclude.split(',')]
51
52         archive = zipfile.ZipFile(path, 'w', zipfile.ZIP_DEFLATED)
53
54         processed = skipped = 0
55         for book in books:
56             if verbose >= 2:
57                 print 'Parsing', book.slug
58             content = getattr(book, field)
59             if not content:
60                 if verbose >= 1:
61                     print self.style.NOTICE('%s has no %s file' % (book.slug, ftype))
62                 skipped += 1
63                 continue
64             archive.write(content.path, str('%s.%s' % (book.slug, ftype)))
65             processed += 1
66         archive.close()
67
68         if not processed:
69             if skipped:
70                 print self.style.ERROR("No books with %s files found" % ftype)
71             else:
72                 print self.style.ERROR("No books found")
73             return
74
75         if verbose >= 1:
76             print "%d processed, %d skipped" % (processed, skipped)
77             print "Results written to %s" % path