hide banner

[edumed.git] / catalogue / management / commands / republish.py

diff --git a/catalogue/management/commands/republish.py b/catalogue/management/commands/republish.py
index 7ead31b..48f9345 100644 (file)
--- a/catalogue/management/commands/republish.py
+++ b/catalogue/management/commands/republish.py
@@ -2,20 +2,45 @@
 # This file is part of EduMed, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
+from optparse import make_option
+
+import librarian
 from django.core.management.base import BaseCommand
 
 
 class Command(BaseCommand):
     help = 'Republishes all lessons.'
 
+    option_list = BaseCommand.option_list + (
+        make_option('--exclude', dest='exclude', metavar="PATH", default=None,
+                    help='PATH to file with excluded lesson slugs.'),
+        make_option('--ignore-incomplete', action='store_true', dest='ignore_incomplete', default=False,
+                    help='Attachments dir path.'),
+        make_option('--dont-repackage', action='store_false', dest='repackage', default=True,
+                    help='Don\'t refresh level packages.'),
+    )
+
     def handle(self, **options):
         from catalogue.models import Lesson
         from curriculum.models import Level
 
-        for lesson in Lesson.objects.all():
-            print 'Republishing: %s' % lesson.title
-            lesson.republish(repackage_level=False)
+        lessons = Lesson.objects.order_by('slug')
+
+        if options.get('exclude'):
+            slugs = [line.strip() for line in open(options['exclude'])]
+            lessons = lessons.exclude(slug__in=slugs)
+
+        for lesson in lessons:
+            print
+            print 'Republishing: %s' % lesson.slug
+            try:
+                lesson.republish(repackage_level=False)
+            except librarian.ParseError as e:
+                print '!!!!!! PARSE ERROR !!!!!!'
+                print e
 
-        print 'Rebuilding levels...'
-        for level in Level.objects.all():
-            level.build_packages()
+        if options.get('repackage'):
+            print 'Rebuilding levels...'
+            for level in Level.objects.all():
+                print level.name
+                level.build_packages()