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                 li_adv = 1
  27                 for lesson in section.lesson_set.all():
  28                     advanced = lesson.level.slug == "liceum"
  29                     section_dir = "%d_%s%s" % (
  30                         si + 1, section.slug,
  31                         " (zaawansowane)" if advanced else "")
  32                     if lesson.type == 'course':
  33                         if advanced:
  34                             ind = li_adv
  35                             li_adv += 1
  36                         else:
  37                             ind = li
  38                             li += 1
  39                         prefix = "%s/%02d_%s/" % (
  40                                 section_dir, ind, lesson.slug,
  41                             )
  42                     elif lesson.type == 'synthetic':
  43                         prefix = "%s/%s (synteza)/" % (
  44                                 section_dir, lesson.slug)
  45                     else:
  46                         prefix = "%s/%s/" % (
  47                                 section_dir, lesson.slug)
  48                     lesson.add_to_zip(zipf, student, prefix)
  49             zipf.close()
  50
  51     def handle(self, **options):
  52         verbose = options.get('verbose')
  53
  54         self.build_package(
  55             os.path.join(settings.MEDIA_ROOT, settings.CATALOGUE_PACKAGE),
  56             False, verbose)
  57         self.build_package(
  58             os.path.join(settings.MEDIA_ROOT, settings.CATALOGUE_PACKAGE_STUDENT),
  59             True, verbose)