Django中删除DeltaStock模型时能否自动执行函数恢复关联Stock的库存？

嘿，这个需求完全可以搞定！Django的信号系统就是专门用来处理这种“某个模型发生特定操作时自动执行逻辑”的场景，不管你是通过代码调用delete()、后台Admin删除，还是其他方式删除DeltaStock实例，都能自动触发库存回加的操作。下面给你详细的实现步骤：

1. 使用post_delete信号处理单个实例删除 #

首先，我们可以利用Django的post_delete信号——这个信号会在DeltaStock实例被成功删除后触发，刚好适合我们在删除完成后回滚库存的需求。

代码实现 #

在你的app的signals.py文件（如果没有就新建一个）里写如下代码：

from django.db.models.signals import post_delete
from django.dispatch import receiver
from .models import DeltaStock, Stock

@receiver(post_delete, sender=DeltaStock)
def rollback_stock_on_deltastock_delete(sender, instance, **kwargs):
    # 回滚库存：把delta_quantity加回关联的Stock的quantity
    instance.stock.quantity += instance.delta_quantity
    instance.stock.save()

然后，你需要在app的apps.py里注册这个信号，确保Django启动时能加载到：

from django.apps import AppConfig

class YourAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'your_app_name'

    def ready(self):
        import your_app_name.signals  # 导入信号模块

最后，记得在settings.py里把你的app配置改成这个自定义的Config：

INSTALLED_APPS = [
    # ...其他app
    'your_app_name.apps.YourAppConfig',
]

2. 处理批量删除的场景 #

上面的信号能处理单个实例的删除，但如果是通过DeltaStock.objects.filter(...).delete()这种批量删除操作，Django默认不会给每个实例发送post_delete信号，这时候我们需要额外处理。

方案：自定义QuerySet和Manager #

我们可以自定义一个QuerySet，重写它的delete()方法，让批量删除时也能逐个处理库存回滚：

from django.db import models

class DeltaStockQuerySet(models.QuerySet):
    def delete(self):
        # 先遍历所有要删除的实例，回滚库存
        for instance in self:
            instance.stock.quantity += instance.delta_quantity
            instance.stock.save()
        # 再执行批量删除
        return super().delete()

class DeltaStockManager(models.Manager):
    def get_queryset(self):
        return DeltaStockQuerySet(self.model, using=self._db)

# 然后修改DeltaStock模型，使用这个自定义Manager
class DeltaStock(models.Model):
    delta_quantity = models.DecimalField(max_digits=10, decimal_places=2)
    stock = models.ForeignKey(to=Stock, on_delete=models.CASCADE)

    objects = DeltaStockManager()

这样不管是单个删除还是批量删除，库存都会被正确回滚啦！

注意事项 #

• 如果你用了事务，记得确保库存回滚和删除操作在同一个事务里，避免出现删除成功但库存没加回去的情况。
• 因为DecimalField的运算要注意精度问题，确保你的quantity和delta_quantity的max_digits和decimal_places设置合理，避免溢出。

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