Skip to content

Commit 2c7d472

Browse files
committed
wrap command in handle() with atomic transaction
1 parent 93fcd18 commit 2c7d472

1 file changed

Lines changed: 7 additions & 5 deletions

File tree

osf/management/commands/copy_collection_submission_metadata_to_cedar.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import logging
22

33
from django.core.management.base import BaseCommand
4+
from django.db import transaction
45

56
from osf.models import CollectionSubmission
67

@@ -66,8 +67,9 @@ def add_arguments(self, parser):
6667
)
6768

6869
def handle(self, *args, **options):
69-
copy_collection_submission_metadata_to_cedar(
70-
dry_run=options['dry_run'],
71-
batch_size=options['batch_size'],
72-
provider_id=options.get('provider_id'),
73-
)
70+
with transaction.atomic():
71+
copy_collection_submission_metadata_to_cedar(
72+
dry_run=options['dry_run'],
73+
batch_size=options['batch_size'],
74+
provider_id=options.get('provider_id'),
75+
)

0 commit comments

Comments
 (0)