catalogue/management/commands/repackage.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of EduMed, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 import os
   6 from optparse import make_option
   7 from django.conf import settings
   8 from django.core.management.base import BaseCommand
   9 from catalogue.models import Section
  10 import zipfile
  11
  12
  13 class Command(BaseCommand):
  14     option_list = BaseCommand.option_list + (
  15         make_option('-q', '--quiet', action='store_false', dest='verbose', default=True,
  16             help='Verbosity level; 0=minimal output, 1=normal output, 2=all output'),
  17     )
  18     help = 'Rebuilds downloadable packages.'
  19
  20     def build_package(self, zippath, student, verbose):
  21         with open(zippath, 'w') as outf:
  22             zipf = zipfile.ZipFile(outf, 'w', zipfile.ZIP_STORED)
  23
  24             for si, section in enumerate(Section.objects.all()):
  25                 li = 1
  26                 for lesson in section.lesson_set.all():
  27                     if lesson.type == 'course':
  28                         prefix = "%d_%s/%02d_%s/" % (
  29                                 si, section.slug,
  30                                 li, lesson.slug,
  31                             )
  32                         li += 1
  33                     elif lesson.type == 'synthetic':
  34                         prefix = "%d_%s/synteza_%s/" % (
  35                                 si, section.slug, lesson.slug)
  36                     else:
  37                         prefix = "%d_%s/%s/" % (
  38                                 si, section.slug, lesson.slug)
  39                     lesson.add_to_zip(zipf, student, prefix)
  40             zipf.close()
  41
  42     def handle(self, **options):
  43         verbose = options.get('verbose')
  44
  45         self.build_package(
  46             os.path.join(settings.MEDIA_ROOT, settings.CATALOGUE_PACKAGE),
  47             False, verbose)
  48         self.build_package(
  49             os.path.join(settings.MEDIA_ROOT, settings.CATALOGUE_PACKAGE_STUDENT),
  50             True, verbose)