Django模型字段的默认值可以由依赖于外部父模型的函数定义吗?

2024-04-29 06:58:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图让默认值Report的费用基于父模型的属性。我不想在save()中执行此操作,因为如果用户选择在保存之前覆盖该值,则需要向用户显示该字段。

除了只传递函数指针(即不传递())之外,我还尝试了以下三种方法。当我运行python manage.py shell时会引发错误。

#1
from django.db import models

class Job(models.Model):
    veryImportant = models.IntegerField()
    def get_fee(self):
        return 2 * self.veryImportant

class Report(models.Model):
    job = models.ForeignKey(Job)
    overridableFee = models.DecimalField(default=job.get_fee(), max_digits=7, decimal_places=2)

#gives...
#AttributeError: 'ForeignKey' object has no attribute 'get_fee'

#2
from django.db import models

class Job(models.Model):
    veryImportant = models.IntegerField()

class Report(models.Model):
    job = models.ForeignKey(Job)
    overridableFee = models.DecimalField(default=self.set_fee(), max_digits=7, decimal_places=2)
    def set_fee(self):
        overridableFee =  2 * self.job.veryImportant

#gives...
#NameError: name 'self' is not defined

#3
from django.db import models

class Job(models.Model):
    veryImportant = models.IntegerField()
    def get_fee():
        return 2 * veryImportant

class Report(models.Model):
    job = models.ForeignKey(Job)
    overridableFee = models.DecimalField(max_digits=7, decimal_places=2)
    def __init__(self, *args, **kwargs):
        self.overridableFee = self.job.get_fee()
        super(models.Model, self).__init__(*args, **kwargs)

#gives...
#TypeError: object.__init__() takes no parameters

3的错误可能是我不知道如何正确重写init。我从另一个答案中抄袭了一些东西,但没用就放弃了。

如果这些都不起作用,我可以在创建视图中的每个报表后恢复为只设置费用,但我宁愿在创建报表时自动设置,以保持头脑清醒。

编辑:

最后得到了#3,用Yuji的答案修复并修改以确保shell设置的任何可能的费用覆盖job.get#fee()想要的。

def __init__(self, *args, **kwargs):
    super(Report, self).__init__(*args, **kwargs)
    if self.overridableFee == None and self.job and not self.pk:
        self.overridableFee = self.job.get_fee()

Tags: selfreportgetmodelinitmodelsdefargs
1条回答
网友
1楼 · 发布于 2024-04-29 06:58:09

最后一个示例可能会处理一些工作:

  1. 首先,您需要在类上__init__,而不是models.Model
  2. 在初始化模型后,需要设置属性
  3. 您需要检查模型是否已保存,否则每次加载时模型都将恢复为可重写的费用。

-

class Job(models.Model):
    veryImportant = models.IntegerField()
    def get_fee():
        return 2 * veryImportant

class Report(models.Model):
    job = models.ForeignKey(Job)
    overridableFee = models.DecimalField(max_digits=7, decimal_places=2)
    def __init__(self, *args, **kwargs):
        super(Report, self).__init__(*args, **kwargs)
        if not self.id:
            self.overridableFee = self.job.get_fee() 

相关问题 更多 >