Django模型删除方法不覆盖

Question

我目前正在开发一个 Django 博客应用程序，其中我有一个 Post 连接到一个 Category。 Category 具有 name 和 times_used 字段。 times_used 将在创建 Post 后增加 1，并且我已经实现了覆盖 Post 模型上的 save() 方法。我正在尝试覆盖 delete() 方法以将当前帖子的类别减 1，但它实际上并没有达到似乎覆盖的 delete() 方法，因为即使在帖子被删除后类别仍然具有增量值。请参阅下面的代码...

model.py

class Category(models.Model):
    name = models.CharField(max_length=150, unique=True)
    times_used = models.IntegerField(default=0)


class Post(models.Model):
    category = models.ForeignKey(Category, on_delete=models.CASCADE)

    # This increments this category by 1 then saves the post - Works well!
    def save(self, *args, **kwargs):
        self.category.times_used += 1
        self.category.save()

        super(Post, self).save(*args, **kwargs)

    # This decrements this category by 1 then deletes the post - Which is not working...
    def delete(self, *args, **kwargs):
        self.category.times_used -= 1
        self.category.save()

        super(Post, self).delete(*args, **kwargs)

有什么建议可以做到这一点吗？

Answer 1

我刚刚阅读了有关 Django 信号的信息，并且已经完成了我想做的事情。根据 Django documentation...，当我们重写它以使用 QuerySet 批量删除对象时，不会调用 delete() 方法。所以最好使用信号来完成这些类型的工作。

在我的models.py 中，我将Post 作为参数传递给我在signals.py 中创建的函数

models.py

class Post(models.Model):
    ....

delete_post_signal(Post)

在我的signals.py 中，我有以下代码...

from django.dispatch import receiver
from django.db.models.signals import pre_delete


def delete_post_signal(Post):

    @receiver(pre_delete, sender=Post)
    def decrement_category_usage(sender, instance, **kwargs):
        instance.category.times_used -= 1
        instance.category.save()

这对我来说非常有用......

Answer 2

你可以像这样使用 pre_delete 信号来做到这一点

@receiver(pre_delete, sender=Post)
def my_handler(sender, instance, **kwargs):
    instance.category.times_used -= 1
    instance.category.save()