如何在Django用户模型中保存时执行代码？

Question

我想专门为Django用户更改is_active字段时运行一些代码，类似于save方法对其他模型的工作方式：

class Foo(models.Model):
    ...
    def save(self, *args, **kwargs):
        if self.pk is not None:
            orig = Foo.objects.get(pk=self.pk)
            if orig.is_active != self.is_active:
                # code goes here

可以通过与用户模型一对一关系的另一个模型来完成此操作吗？ 就像是：

class Bar(models.Model):
    owner = models.OneToOneField(User, on_save=?)
    ...

我想我可以在相关模型上复制is_active字段，然后在保存相关模型时在User上设置is_active字段。 但这似乎有点麻烦。

Answer 1

您正在寻找这个信号

from django.db.models.signals import pre_save
from django.contrib.auth.models import User

def do_your_thing(sender, instance, **kwargs):
    # Do something
    print(instance)

pre_save.connect(do_your_thing, sender=User)