refactor: Add atomicity to function calls in execute

openedx · Apr 10, 2023 · 54ea975 · 54ea975
1 parent b9f921b
commit 54ea975
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ecommerce/extensions/iap/api/v1/views.py b/ecommerce/extensions/iap/api/v1/views.py
@@ -214,14 +214,16 @@ def post(self, request):
             return JsonResponse({'error': ERROR_DURING_PAYMENT_HANDLING}, status=400)
 
         try:
-            order = self.create_order(request, basket)
+            with transaction.atomic():
+                order = self.create_order(request, basket)
         except Exception:  # pylint: disable=broad-except
             # Any errors here will be logged in the create_order method. If we wanted any
             # IAP specific logging for this error, we would do that here.
             return JsonResponse({'error': ERROR_DURING_ORDER_CREATION}, status=400)
 
         try:
-            self.handle_post_order(order)
+            with transaction.atomic():
+                self.handle_post_order(order)
         except Exception:  # pylint: disable=broad-except
             self.log_order_placement_exception(basket.order_number, basket.id)
             return JsonResponse({'error': ERROR_DURING_POST_ORDER_OP}, status=200)