Aborting transaction context manager.

aborting_transaction.py

"""
Aborting transaction context manager, useful for when prototyping on Django.
"""
from contextlib import contextmanager
from django.db import transaction

class AbortTransaction(Exception):
    pass

@contextmanager
def aborting_transaction():
    try:
        with transaction.atomic():
            yield
            raise AbortTransaction()
    except AbortTransaction:
        pass