在调用 Django 模型保存/删除后调用函数

Question

我在 Django 中有很多模型类。它们都没有被覆盖的 save() 或 delete() 函数。

我想创建一个类级别的装饰器，用于在调用 save() / delete() 后执行一个函数。这个类级别的装饰器应该是通用的，以便所有模型类都可以使用它。

我已经尝试使用 post_save 和 post_delete 的信号，但是维护大量的类列表可能是个问题。我也想避免覆盖 save()/delete()。

关于如何去做的任何建议？提前致谢。

Answer 1

推荐使用 post_save 信号。

如果您不想单独注册每个模型类，这里有一个使用类装饰器的解决方案：

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

class SaveRegister:
    registered_classes = []
    post_save_func = None

    @classmethod
    def register(cls, target_cls):
        if not cls.post_save_func:
            raise Exception("Post save function not defined")
        if target_cls not in cls.registered_classes:
            # prevent duplicate registration
            cls.registered_classes.append(target_cls)
            # register the model class to listen to post_save
            receive(post_save, sender = target_cls)(cls.post_save_func)

    # set the post_save signal handler
    @classmethod
    def set_post_save_func(cls, f):
        cls.post_save_func = f

现在您所要做的就是定义一个 post_save 处理程序，将其设置到 SaveRegister 类中：

def model_post_save(sender, **kwargs):
    print('Saved an instance with type: {}'.format(sender))
SaveRegister.set_post_save_func(model_post_save)

每个用SaveRegister.register 修饰的模型类都会自动拥有一个 post_save 处理程序：

@SaveRegister.register
class Student(SignaledModel):
    number = models.PositiveIntegerNumber(unique=True)
    name = models.CharField(max_length=64)

@SaveRegister.register
class Course(SignaledModel):
    code = models.CharField(unique=True)
    teacher = models.CharField(max_length=64)

Answer 2

您是否尝试过创建 abstract 模型类并从中继承所有实际模型？这样，当您将pre_save 信号附加到基类时，只要保存了任何子模型，就会发送该信号。

class SignaledModel(models.Model):
    class Meta:
        abstract = True

class Student(SignaledModel):
    number = models.PositiveIntegerNumber(unique=True)
    name = models.CharField(max_length=64)

class Course(SignaledModel):
    code = models.CharField(unique=True)
    teacher = models.CharField(max_length=64)

---

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

@receiver(post_save, sender=SignaledModel)
def model_post_save(sender, **kwargs):
    print('Saved an instance with type: {}'.format(sender))

每当保存Course 或Student 实例时，都会调用model_post_save。

希望这会有所帮助。