【问题标题】:Django Model Save Override Simpler Approach?Django模型保存覆盖更简单的方法?
【发布时间】:2015-07-15 20:50:18
【问题描述】:

我希望覆盖模型的保存方法,以便它在保存之前进行一些简单的计算,然后保存最终结果。计算示例(来自我的模型):

total_ticket = ticket2 - ticket1
gross_sales = rate * total_ticket
total_tax = tax * total_ticket

虽然我已经能够完成它,但我很想知道是否有其他更简单的方法可以完成同样的任务。

看看我的models.py,你就会明白我的意思了

class Entry(models.Model):
    ticket1 = models.IntegerField()
    ticket2 = models.IntegerField()
    total_ticket = models.IntegerField(default=0, editable=False)
    rate = models.IntegerField(null=True, blank=True)
    tax = models.IntegerField(null=True, blank=True)
    gross_sales = models.IntegerField(default=0, editable=False)
    total_tax = models.IntegerField(default=0, editable=False)

    def save(self, *args, **kwargs):
        # calculate sum before saving.
        self.total_ticket = self.calculate_ticket()
        self.gross_sales = self.calculate_gross_sales()
        self.total_tax = self.calculate_tax()
        super(Entry, self).save(*args, **kwargs)

    def calculate_gross_sales(self):
        try:
            total_ticket = self.total_ticket
            rate = self.rate
            return total_ticket * rate
        except KeyError:
            return 0

    def calculate_tax(self):
        try:
            total_ticket = self.total_ticket
            tax = self.tax
            return total_ticket * tax
        except KeyError:
            return 0

    def calculate_ticket(self):
        """ Calculate a numeric value for the model instance. """
        try:
            ticket1 = self.ticket1
            ticket2 = self.ticket2
            return ticket2 - ticket1
        except KeyError:
            return 0

    def __str__(self):
        return self.show

【问题讨论】:

  • 这应该没问题,但是如果您正在寻找抽象,您可以查看pre_save 信号

标签: python django


【解决方案1】:

正如@karthinkr 在 cmets 中所说,您可以使用 pre_save。举个例子:

from django.db.models.signals import pre_save

def calculate_things(sender, instance, **kwargs):
    # sender => Model
    # instance => Object pre saved
    ...
    # Your loginc here

class Entry(models.Model):
    ...
    # Model definition
    ...

    ... # You don't need re define save method.


pre_save.connect(calculate_things, sender=Entry)

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2018-10-07
    • 1970-01-01
    • 2021-08-11
    • 2020-08-07
    • 2016-07-26
    • 2020-06-14
    • 1970-01-01
    相关资源
    最近更新 更多