transaction_middleware.py

class TransactionMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response
        
    def __call__(self, request):
        with transaction.atomic():
            response = self.get_response(request)
            if response.status_code == 500:
                transaction.set_rollback()
        return response