newsletter stub

[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 216c5e3..6196bab 100755 (executable)
--- a/src/catalogue/management/commands/savemedia.py
+++ b/src/catalogue/management/commands/savemedia.py
@@ -5,25 +5,21 @@
 import os.path
 
 from django.core.management.base import BaseCommand
 import os.path
 
 from django.core.management.base import BaseCommand

+from django.db import transaction

 
 from catalogue.models import Book, BookMedia
 from catalogue.utils import ExistingFile
 
 
 class Command(BaseCommand):
 
 from catalogue.models import Book, BookMedia
 from catalogue.utils import ExistingFile
 
 
 class Command(BaseCommand):

-    help = "Saves uploaded media with a given book and a given name. If media has a source SHA1 info - matching media is replaced."
+    help = "Saves uploaded media with a given book and a given name. " \
+           "If media has a source SHA1 info - matching media is replaced."

     args = 'path slug name'
 
     args = 'path slug name'
 

+    @transaction.atomic

     def handle(self, *args, **options):
     def handle(self, *args, **options):

-        from django.db import transaction
-

         path, slug, name = args
 
         path, slug, name = args
 

-        # Start transaction management.
-        transaction.commit_unless_managed()
-        transaction.enter_transaction_management()
-        transaction.managed(True)
-

         book = Book.objects.get(slug=slug)
 
         root, ext = os.path.splitext(path)
         book = Book.objects.get(slug=slug)
 
         root, ext = os.path.splitext(path)


@@ -45,5 +41,3 @@ class Command(BaseCommand):
         bm.name = name
         bm.file.save(None, ExistingFile(path))
         bm.save()
         bm.name = name
         bm.file.save(None, ExistingFile(path))
         bm.save()

-        transaction.commit()
-        transaction.leave_transaction_management()