add option to choose cover class

[wolnelektury.git] / src / catalogue / management / commands / savemedia.py

diff --git a/src/catalogue/management/commands/savemedia.py b/src/catalogue/management/commands/savemedia.py
index 6196bab..ab4da51 100755 (executable)
--- a/src/catalogue/management/commands/savemedia.py
+++ b/src/catalogue/management/commands/savemedia.py
@@ -18,7 +18,9 @@ class Command(BaseCommand):
 
     @transaction.atomic
     def handle(self, *args, **options):
 
     @transaction.atomic
     def handle(self, *args, **options):

-        path, slug, name = args
+        path, slug, name, part_name, index, parts_count = args
+        index = int(index)
+        parts_count = int(parts_count)

 
         book = Book.objects.get(slug=slug)
 
 
         book = Book.objects.get(slug=slug)
 


@@ -39,5 +41,7 @@ class Command(BaseCommand):
             bm = BookMedia(book=book, type=ext)
             print "Creating new media"
         bm.name = name
             bm = BookMedia(book=book, type=ext)
             print "Creating new media"
         bm.name = name

+        bm.part_name = part_name
+        bm.index = index

         bm.file.save(None, ExistingFile(path))
         bm.file.save(None, ExistingFile(path))

-        bm.save()
+        bm.save(parts_count=parts_count)