Django是否存在数据库入库后触发的信号？post_save问题解法

关于内置信号的触发时机说明 #

Django 原生没有提供数据库事务提交完成、数据持久化落库后触发的内置信号。
根据官方定义，post_save 信号的触发时机是模型实例 save() 方法执行到末尾的节点，此时如果 save 操作被包裹在未提交的数据库事务中，数据仅存在于当前数据库连接的事务上下文里，并未持久化到磁盘，跨连接查询、外键约束校验时就会出现「关联主键不存在」的报错，这就是在模型A的post_save中创建关联外键的模型B时触发报错的根本原因。

落地实现方案：transaction.on_commit() 正确用法 #

不需要自定义新的信号，直接在post_save信号回调中通过transaction.on_commit()将需要延后执行的逻辑绑定到事务提交成功的节点即可，使用规则如下：

• 导入事务模块：from django.db import transaction
• 需要延后执行的逻辑封装为无参回调函数，通过闭包捕获当前信号中的模型实例、created状态等变量，避免后续引用变动导致逻辑异常
• on_commit绑定的回调仅会在当前事务成功提交后执行，如果事务发生回滚，回调会被直接丢弃，不会执行
• 该方法会自动适配Django的两种运行模式：默认自动提交模式下（AUTOCOMMIT=True），回调会在当前SQL执行完成后立即触发；在transaction.atomic()显式开启的事务块中，回调会等待整个事务块所有操作执行完成、正式提交后才会触发

代码示例 #

from django.db import transaction
from django.db.models.signals import post_save
from django.dispatch import receiver
from .models import ModelA, ModelB

@receiver(post_save, sender=ModelA)
def handle_modela_created(sender, instance, created, **kwargs):
    # 按需判断是否是首次创建实例，更新场景可直接跳过
    if not created:
        return

    def _create_related_b():
        # 该段逻辑会在ModelA数据正式落库后执行
        ModelB.objects.create(a_id=instance.pk)

    # 绑定回调到事务提交节点
    transaction.on_commit(_create_related_b)

常见注意事项 #

• 不要在on_commit回调中依赖未提交的事务上下文，回调执行时原事务已经结束，所有需要用到的实例参数要提前在闭包中捕获
• 单元测试场景下，Django默认的TestCase会为每个用例包裹事务并在结束后回滚，此时on_commit绑定的回调不会自动执行，需要用captureOnCommitCallbacks上下文管理器捕获回调手动触发，或者改用TransactionTestCase编写相关用例
• 不要给on_commit传入带参数的回调函数，如果需要传参统一通过闭包或者functools.partial封装为无参函数后传入

内容的提问来源于stack exchange，提问作者Harshit Joshi