Run migration with atomic transactions

ExCiteS · May 19, 2019 · 6dfd189 · 6dfd189
1 parent 55e9ae0
commit 6dfd189
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/geokey/contributions/migrations/0020_update_media_and_comments_count.py b/geokey/contributions/migrations/0020_update_media_and_comments_count.py
@@ -1,16 +1,17 @@
 # -*- coding: utf-8 -*-
 
-from django.db import migrations, IntegrityError
+from django.db import migrations, transaction, IntegrityError
 
 
 def update_media_and_comments_count(apps, schema_editor):
     Observation = apps.get_model('contributions', 'Observation')
 
     for observation in Observation.objects.all():
         try:
-            observation.num_media = observation.files_attached.count()
-            observation.num_comments = observation.comments.count()
-            observation.save()
+            with transaction.atomic():
+                observation.num_media = observation.files_attached.count()
+                observation.num_comments = observation.comments.count()
+                observation.save()
         except IntegrityError:
             pass