Django transaction.atomic()事务处理

在Django中,transaction.atomic()是一个上下文管理器,它会自动开始一个事务,并在代码块执行完毕后提交事务。如果在代码块中抛出异常,事务将被自动回滚,确保数据库的一致性和完整性。

在实际应用中,你可能需要根据具体的业务逻辑来决定何时抛出异常,以触发事务回滚。例如,如果某个操作依赖于另一个操作的成功,那么在第一个操作失败时,你可能需要抛出异常,以防止后续操作被执行,从而保持数据的一致性。

1,添加模型

Test/app11/models.py

复制代码
from django.db import models

class Post(models.Model):
    title = models.CharField(max_length=200)
    content = models.TextField()
    pub_date = models.DateTimeField('date published')




class Book(models.Model):
    title = models.CharField(max_length=100)
    author = models.CharField(max_length=100)
    publication_date = models.DateField()
    price = models.DecimalField(max_digits=5, decimal_places=2)

    def __str__(self):
        return self.title

2,添加视图函数

Test/app11/views.py

2.1 装饰器写法

复制代码
from django.http import HttpResponse
from django.db import transaction
from .models import Post, Book
from django.utils import timezone


def atomic_transaction(func):
    def wrapper(*args, **kwargs):
        with transaction.atomic():
            try:
                return func(*args, **kwargs)
            except Exception as e:
                # 在这里可以添加错误日志记录或其他错误处理逻辑
                print(f"An error occurred: {e}")
                # 由于在transaction.atomic()上下文中,异常将自动触发事务回滚

    return wrapper


@atomic_transaction
def create_post_and_book(request):
    post_title = "New Post4"
    book_title = "New Book4"

    post = Post(title=post_title, content="This is the content of the new post.", pub_date=timezone.now())
    post.save()

    book = Book(title=book_title, author="Author Name", publication_date=timezone.now().date(), price=29.99)
    book.save()

    # 故意引发异常以测试回滚
    # if True:
    #     raise Exception("An error occurred")

    print("Post and Book created successfully.")

    return HttpResponse("Post and Book created successfully.")

2.2 with写法

复制代码
from django.http import HttpResponse
from django.db import transaction
from .models import Post, Book
from django.utils import timezone


def create_post_and_book(request):
    post_title = "New Post3"
    book_title = "New Book3"

    with transaction.atomic():
        post = Post(title=post_title, content="This is the content of the new post.", pub_date=timezone.now())
        post.save()

        # 故意引发异常以测试回滚
        if True:
            raise Exception("An error occurred")

        book = Book(title=book_title, author="Author Name", publication_date=timezone.now().date(), price=29.99)
        book.save()

    return HttpResponse("Post and Book created successfully.")

3,添加路由应用

Test/app11/urls.py

复制代码
from django.urls import path
from . import views

urlpatterns = [

    path('create_post_and_book/', views.create_post_and_book, name='create_post_and_book'),
]

4,访问页面

http://127.0.0.1:8000/app11/create_post_and_book/

当你的代码中抛出异常,如raise Exception("An error occurred"),这会中断程序的正常执行流程。在Django的事务处理中,如果在transaction.atomic()的上下文中抛出异常,Django会捕获这个异常并自动回滚事务,确保任何在事务开始后所做的数据库更改不会被永久保存。

可以看到数据并没有被保存到

在这个例子中,由于if True:条件总是为真,因此raise Exception("An error occurred")这一行代码将总是抛出异常。结果,transaction.atomic()会捕获这个异常并回滚事务,这意味着Post和Book对象都不会被保存到数据库中,因为事务没有成功提交。这是Django事务处理的一个关键特性,它确保了数据的一致性和完整性,避免了部分完成的事务可能导致的数据库状态不一致问题。

相关推荐
JaydenAI1 分钟前
[对比学习LangChain和MAF-07]如何引入人机交互的审批流程
python·ai·langchain·c#·agent·hitl·maf
神奇元创39 分钟前
商用级光路加速卡:大模型推理的极速落地方案
python·神经网络·fpga开发·dsp开发
运筹vivo@1 小时前
Python ContextVar 底层机制与内存模型拆解
前端·数据库·python
大白菜和MySQL1 小时前
java应用排查高线程
java·python
嵌入式协会20240721 小时前
(已解决)MinIO python 获取预签名出现forbidden、errornetwork等错误
java·开发语言·python
宸丶一1 小时前
Day 14:任务追踪 - 让 Agent 拥有项目管理能力
开发语言·python
skylar02 小时前
小白1分钟安装flash-attn
开发语言·python
JustNow_Man2 小时前
psmux快捷键
人工智能·python
默子昂2 小时前
ollama 自定义ui
开发语言·python·ui