保存时对相关模型执行操作

Posted 2023-02-22

【中文标题】保存时对相关模型执行操作

【英文标题】：Performing actions on a related model while saving

【发布时间】：2021-06-14 17:59:42

【问题描述】：

我需要帮助。几天来，我一直无法弄清楚如何解决一个问题。我将不胜感激任何想法或建议。 为简单起见，有一组模型：

class A(models.Model):
    name = models.CharField(max_length=255)

class B(models.Model):
    name = models.CharField(max_length=255)
    owner = models.ForeignKey(A, on_delete=models.CASCADE, null=False)

def foo():
    do something

事先不知道属于A的B的实例数。 B 的实例数由用户在创建新的 A 实例时确定。此外，他可以创建 A 的实例而无需创建 B 的单个实例（空集）。

挑战：我需要在保存 A 的新实例后运行 foo。此函数应同时处理 A 的新实例的字段和 B 的字段（反向关系）。

是的，我已将接收器设置为 post_save 信号模型 A。但问题是，此时 A 已被保存（post_save em>)，B的实例还没有被保存(pre_save)，对应字段的值还没有确定。换句话说，我无法在接收器 A 中获取原始 B 值。

有什么想法吗？我做错了吗？

【问题讨论】：

如果您需要在某些其他功能发生时发生特定功能，则需要在应用程序中实现信号。为此，您特别需要 post_save 方法。这是一个链接，可以了解有关信号如何工作的更多信息docs.djangoproject.com/en/3.1/topics/signals

还告诉我们有关函数 foo 的更多信息，以便我们提供解决方案。

感谢您的回答！当然，我已经实现了 post_save 信号接收器。但问题是，在信号到达的时候，还没有 B 的实例（A.b.None）实际上，这个函数应该动态创建一个名为 A.name 和字段集 B 的新模型

【参考方案1】：

我找到了解决方案。也许它对某人有用。 我听取了我们更有经验的同事的建议，将信号仅作为最后的手段。结果证明解决方案很简单。虽然解决方案可能看起来不是很优雅。

解决方案是基于覆盖每个模型的 save() 方法

class A(models.Model):
    name = models.CharField(max_length=255)
 
def _call_foo(self,  *args, **kwargs):
    do something
    foo(*args, **kwargs)
 
def save(self, *args, **kwargs):
    changed_fields = []
    pk = None
    if self.pk:
        cls = self.__class__
        old = cls.objects.get(pk=self.pk)
        new = self
        for field in cls._meta.get_fields():
            field_name = field.name
            try:
                if getattr(old, field_name) != getattr(new, field_name):
                    pk = self.pk
                    changed_fields.append(field_name)
            # Catch field does not exist exception
            except Exception as ex:
                print(type(ex))
                print(ex.arg)
                print(ex)
        kwargs['update_fields'] = changed_fields
        super().save(*args, **kwargs)
    else:
        super().save(*args, **kwargs)
        pk = self.pk
        changed_fields.append('__all__')
    self._call_foo(self, pk, changed_fields)
 
class B(models.Model):
    name = models.CharField(max_length=255)
    owner = models.ForeignKey(A, on_delete=models.CASCADE, null=False) 
 
    def save(self, *args, **kwargs):
    changed_fields = []
    pk = None
    if self.pk:
        cls = self.__class__
        old = cls.objects.get(pk=self.pk)
        new = self
        for field in cls._meta.get_fields():
            field_name = field.name
            try:
                if getattr(old, field_name) != getattr(new, field_name):
                    pk = self.pk
                    changed_fields.append(field_name)
            # Catch field does not exist exception
            except Exception as ex:
                print(type(ex))
                print(ex.arg)
                print(ex)
        kwargs['update_fields'] = changed_fields
        super().save(*args, **kwargs)
    else:
        super().save(*args, **kwargs)
        pk = self.pk
        changed_fields.append('__all__')
    self.owner._call_foo(self, pk, changed_fields)

此解决方案比使用信号更好，至少在真正需要时调用 save 方法：要么创建实例，要么更新它。

当然，需要记住的是，如果用户更新或创建了两个模型的多个实例（例如，在内联表单中，管理员创建了多个 B 实例和一个 A 实例），那么 foo( ) 函数将被调用多次。但这已经可以在 _call_foo() 方法级别解决，也可以通过 foo() 函数本身的逻辑来解决。