Utilize Transaction Management for case where Data already exists (#1623

)
refinery-platform · Mar 1, 2017 · d48e1e1 · d48e1e1
1 parent b01ad23
commit d48e1e1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/refinery/core/migrations/0005_ontology_data.py b/refinery/core/migrations/0005_ontology_data.py
@@ -77,7 +77,8 @@ def forwards(apps, schema_editor):
         # Attempt to save Ontology objects. If an IntegrityError is raised
         # upon saving an object we continue as it already exists.
         try:
-            ontology.save()
+            with transaction.atomic():
+                ontology.save()
         except IntegrityError:
             continue
 

diff --git a/refinery/file_store/migrations/0002_filetypes_and_fileextensions.py b/refinery/file_store/migrations/0002_filetypes_and_fileextensions.py
@@ -300,7 +300,8 @@ def forwards(apps, schema_editor):
         # Attempt to save FileExtension objects. If an IntegrityError is raised
         # upon saving an object we continue as it already exists.
         try:
-            file_extension.save()
+            with transaction.atomic():
+                file_extension.save()
         except IntegrityError:
             continue