Question

我在Django中有很多模型类。其中没有一个具有重写的save（）或delete（）函数。

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

我已经尝试过使用post_save和post_delete的信号，但维护一个庞大的类列表可能会有问题。我想避免覆盖save（）/ delete（）。

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

Answer 1

您是否尝试创建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。

希望这有帮助。

Answer 2

建议使用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)

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

2 个答案: