Django - 基于另一个字段更新模型字段

Question

我是 Django 和 Python 的新手，我想做一些我以前在 Java EE 中经常做的事情。

考虑以下模型（仅相关类）：

class Item(models.Model):
    name = models.CharField(max_length=40)
    default_price = models.DecimalField(max_digits=6, decimal_places=2, default=50)

    def __unicode__(self):
        return self.name


class SaleDetail(models.Model):
    item = models.ForeignKey(Item)
    deposit = models.ForeignKey(Deposit)
    quantity = models.PositiveIntegerField()
    unit_price = models.DecimalField(max_digits=6, decimal_places=2)
    sale = models.ForeignKey(Sale)

    def item_updated(self, value):
        if self.unit_price == None:
            self.unit_price = value

我想要做的是每次将Item 添加到SaleDetail 时，如果SaleDetail 是新的或未设置unit_price，则将SaleDetail.unit_price 更新为Item.default_price。

我过去在 Java POJO 中所做的是将这个逻辑包含在 setter 方法中。我尝试使用 python 属性来封装 item 属性，但是 Django 直接在后台更新了该字段，因此这会破坏一些自动功能。

我也尝试子类化 ForeignKey 以接受回调函数，但我找不到在容器类上调用方法的方法。

我想这样做，以便为 UI 提供默认值，但我不想在视图逻辑中包含此逻辑，因为从概念上讲，我认为此逻辑应该在模型上（服务器端）

这种情况下的其他用途是更新每个销售细节和销售的总计。我想在用户决定保存销售之前计算这个，所以保存信号不起作用。

谢谢！

Answer 1

看看这是否适合你。我们重写 save 方法并检查是否有pk。如果pk 为None，我们知道SaleDetail 是新的。然后，我们看看是否有unit_price。如果没有unit_price，我们将其设置为Item.default_price。

class Item(models.Model):
    name = models.CharField(max_length=40)
    default_price = models.DecimalField(max_digits=6, decimal_places=2, default=50)

    def __unicode__(self):
        return self.name


class SaleDetail(models.Model):
    item = models.ForeignKey(Item)
    deposit = models.ForeignKey(Deposit)
    quantity = models.PositiveIntegerField()
    unit_price = models.DecimalField(max_digits=6, decimal_places=2)
    sale = models.ForeignKey(Sale)

    def save(self, *args, **kwargs):
        # Make sure this is the first save (pk should be None) and there is no unit_price set
        if self.pk is None and not self.unit_price:
            self.unit_price = self.item.default_price
        elif not self.unit_price:
            self.unit_price = self.item.default_price

        # Call the original save method
        super(SaleDetail, self).save(*args, **kwargs)

Answer 2

“单价”实际上是两个不同字段的函数。因此，我倾向于这样写：

class Item(models.Model):
    name = models.CharField(max_length=40)
    default_price = models.DecimalField(max_digits=6, decimal_places=2, default=50)

    def __unicode__(self):
        return self.name

class SaleDetail(models.Model):
    item = models.ForeignKey(Item)
    deposit = models.ForeignKey(Deposit)
    quantity = models.PositiveIntegerField()
    entered_unit_price = models.DecimalField(max_digits=6, decimal_places=2, default=None)
    sale = models.ForeignKey(Sale)

    @property
    def unit_price(self, value):
        if self.entered_unit_price is None:
            return self.item.default_price
        else:
            return self.entered_unit_price

    @unit_price.setter
    def unit_price(self, value):
        self.entered_unit_price = value

然后像这样使用它：

print(sd.unit_price)
sd.unit_price = 500 
sd.save()