【问题标题】:Django: Custom Save method for Many-to-Many relationDjango:多对多关系的自定义保存方法
【发布时间】:2012-08-12 21:47:12
【问题描述】:

我需要在多对多关系上设置自定义保存和删除方法。

我尝试使用“through”属性指定模型,但这使我的代码过于复杂并引入了一些问题。我不需要多对多模型上的任何额外字段,只需自定义保存和删除方法即可。

是否可以在不指定“through”属性的情况下完成此操作?

代码如下:

class Order(BaseDate):
    #lots of fields
    relateds = models.ManyToManyField('RelatedProduct', verbose_name=_('related products'), blank=True, related_name='order_relateds', through='OrderRelateds')
    # more fields
    total = CurrencyField(verbose_name=_('total'))

    def calculate_total(self):
        cleanses = self.cleanse.taxed_price() * self.quantity
        delivery = DELIVERY_PRICE if self.delivery == 'delivery' else 0
        relateds = 0
        for r in self.relateds.all():
            relateds = relateds + float(r.taxed_price())
        total = float(cleanses) + delivery + relateds
        return total

    def save(self, *args, **kwargs):
        self.total = '%.2f' % self.calculate_total()
        super(Order, self).save(*args, **kwargs)

class OrderRelateds(models.Model):
    order = models.ForeignKey(Order)
    relatedproduct = models.ForeignKey(RelatedProduct, verbose_name=_('related product'))

    class Meta:
        verbose_name = _('Related Product')
        verbose_name_plural = _('Products Related to this Order')

    def __unicode__(self):
        return self.relatedproduct.__unicode__()

    def save(self, *args, **kwargs):
        super(OrderRelateds, self).save(*args, **kwargs)
        self.order.save()

    def delete(self, *args, **kwargs):
        super(OrderRelateds, self).delete(*args, **kwargs)
        self.order.save()

如果在订单中添加或删除任何相关产品(多对多项目),我需要触发订单总价的重新计算。

编辑:这是解决我问题的代码

from django.db.models.signals import m2m_changed
from django.dispatch import receiver

@receiver(m2m_changed, sender=Order.relateds.through)
def recalculate_total(sender, instance, action, **kwargs):
    """
    Automatically recalculate total price of an order when a related product is added or removed
    """
    if action == 'post_add':
        instance.save()
    if action == 'post_remove' or action == 'post_clear':
        instance.save()

【问题讨论】:

  • 如果您不指定为什么需要自定义保存和删除以及那里应该发生什么,这真的很难回答。两种模型的代码也会有所帮助。

标签: python django django-models django-orm


【解决方案1】:

您可以在包含关系的模型上使用 django 的 m2m_changed, pre_save/post_save, pre_delete/post_delete signals,并在那里执行相关逻辑。

【讨论】:

  • 非常感谢您快速而直接的回答。我将附上解决我问题的代码。
【解决方案2】:

蒂米的回答绝对正确且有效。但是,对于您的特殊情况,我想知道,如果您不应该在视图中处理该逻辑,您可以在其中管理订单和订单项目并在那里执行重新计算,因为订单项目的添加、编辑和删除都属于一起并属于到订单。

【讨论】:

  • 重新计算总需求也可以在管理员中工作以防万一。
猜你喜欢
  • 2015-07-02
  • 2017-12-16
  • 2018-07-05
  • 1970-01-01
  • 1970-01-01
  • 2015-02-03
  • 1970-01-01
  • 2021-12-26
  • 2014-11-21
相关资源
最近更新 更多