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事务处理的一个关键特性,它确保了数据的一致性和完整性,避免了部分完成的事务可能导致的数据库状态不一致问题。

相关推荐
拖孩14 分钟前
AI 帮我写的小程序「厨菜记」上线满月了,挣了两百块!!!
前端·后端·微信小程序
滴啦嘟啦哒18 分钟前
【具身笔记】【π0.5 微调与仿真】五、在MuJoCo中实现可视化仿真
笔记·python·vla·具身
流烟默19 分钟前
SSE 实战:在 SpringBoot + Vue 项目中零依赖实现轻量级实时消息推送
vue.js·spring boot·后端·sse
upala24 分钟前
【仓颉语言入门 · 第二篇 第 9 课】
后端
全栈练习生33 分钟前
AI安全与红队测试
python·ai
费曼学习法39 分钟前
把「发送成功」当「送达」?我排查 IM 长消息被静默吞掉的完整过程
python·ai编程·自动化运维
鸽鸽程序猿41 分钟前
【项目】【在线判题系统】b端题目增删改
开发语言·python
NetX行者1 小时前
Python 工程化入门:从脚本到规范的 Python 项目结构
网络·python
Rain的Java大神之路1 小时前
别再死磕 MySQL 了:向量数据库与 RAG 生产落地实录(Java 工程师专场)
java·后端·面试
MasonYyp1 小时前
asyncio并发协程
python