在保存到 Django admin 之前使用外键模型中的数据

Question

我正在制作一个估算应用程序的原型。

每个 Estimate 模型将有一个或多个 Part 模型。

这是我的模型的开始：

class Part(models.Model):
    estimate = models.ForeignKey('Estimate', on_delete=models.CASCADE)
    description = models.CharField(max_length=255)


class Estimate(models.Model):
    number = models.CharField(max_length=255, primary_key=True)
    customer = models.CharField(max_length=255, null=True, blank=True)
    description = models.CharField(max_length=255, null=True, blank=True)
    price = models.IntegerField(null=True, blank=True)
    stitch = models.BooleanField(null=True, blank=True)
    stitch_speed = models.IntegerField(null=True, blank=True)
    stitch_rate = models.IntegerField(null=True, blank=True)
    boxes = models.IntegerField(null=True, blank=True)
    freight = models.IntegerField(null=True, blank=True)
    overall_markup = models.IntegerField(null=True, blank=True)
    date_created = models.DateField(auto_now_add=True)

    def __str__(self):
        return self.number

我的 Estimate 模型出现在管理员中，而 Part 模型位于 StackedInline 中。

class PartInline(admin.StackedInline):
    model = Part
    extra = 1


class EstimateAdmin(admin.ModelAdmin):
    model = Estimate
    inlines = [PartInline]
    list_display = ['number', 'description', 'customer', 'price']

理想情况下，我会进入管理员，创建一个新的 Estimate，填写其部分，然后单击保存。

最终，我的 Estimate 模型将需要使用 Part 模型中的数据来执行计算，然后再保存到数据库中。例如，每个零件都包含自己的价格。每个零件的价格都需要添加到总估价中。

我无法弄清楚如何在保存之前将数据从 Part 模型发送到 Estimate 模型。

有什么建议吗？是否有在保存之前在模型之间传递数据的标准方法？

Answer 1

据我了解，您可以使用 pre_save 信号，您可以在其中获取与您的 Estimate 实例相关联的所有部分。比如：

from django.dispatch import receiver
from django.db.models.signals import pre_save
from somewhere import Estimate

@receiver(pre_save, sender=Estimate)
def pre_save_handler(sender, instance, **kwargs):
    # instance is your Estimate object that you currently save
    # you can get all part objects with:
    parts = instance.part_set.all() # a QuerySet list with your parts
    for part in parts:
        # do something with each part

编辑：为了使用_set 函数获取所有部分，您必须反转逻辑，以便Part 实例在Estimate 之前通过覆盖EstimateAdmin 的save_formset 保存功能：

class EstimateAdmin(admin.ModelAdmin):
    ....
    def save_formset(self, request, form, formset, change):
        formset.save() # this saves your parts
        form.instance.save() # this saves the Estimate