Django模型继承层次结构中字段的基于类的默认值

Question

如何在以下方案中实现基于类的默认值？我的意思是，我想继承类为“数字”设置不同的默认值：

class OrderDocumentBase(PdfPrintable):
    number = models.PositiveIntegerField(default=self.create_number())

    @classmethod
    def create_number(cls):
        raise NotImplementedError

class Invoice(OrderDocumentBase):
    @classmethod
    def create_number(cls):
        return 1

class CreditAdvice(OrderDocumentBase):
    @classmethod
    def create_number(cls):
        return 2

我查看了this stackoverflow question，但它并没有解决同样的问题。我认为唯一可行的就是像这样重载OrderDocumentBase 的__init__ 方法：

def __init__(self, *args, **kwargs):
    """
    Overload __init__ to enable dynamic set of default to number
    """
    super(OrderDocumentBase, self).__init__(*args, **kwargs)
    number_field = filter(lambda x: x.name == 'number', self._meta.fields)[0]
    number = self.__class__.create_number()
    number_field.default = number

这有效，但只是部分有效，而且表现得很奇怪。在管理界面中，我可以看到仅在第二个或后一页刷新后才设置默认值。第一次尝试时，正在设置None :(

第二种可能性是在每个类中重新定义数字字段，但这似乎不太漂亮。有没有其他办法？

有人可以帮忙吗？

Answer 1

这里有几个问题。首先，self.method 不起作用。在类主体的上下文中没有 self，这是您声明 PositiveIntegerField 的地方。

其次，传递可调用对象将不起作用，因为可调用对象在编译时被绑定并且在运行时不会更改。所以如果你定义说，

class OrderDocumentBase(PdfPrintable):
    create_number = lambda: return 0
    number = models.PositiveIntegerField(default=create_number)

class Invoice(OrderDocumentBase):
    create_number = lambda: return 1

所有Invoice 实例仍将0 作为默认值。

我能想到的解决这个问题的一种方法是覆盖save() 方法。您可以检查是否未提供 number 并将其设置为默认值，然后再保存。

class OrderDocumentBase(PdfPrintable):
    number = models.PositiveIntegerField()

    def save(self, *args, **kwargs):
        if not self.number:
            self.number = self.DEFAULT
        super(OrderDocumentBase, self).save(*args, **kwargs)

class Invoice(OrderDocumentBase):
    DEFAULT = 2

class CreditAdvice(OrderDocumentBase):
    DEFAULT = 3

我对上述内容进行了一些小改动（由于我没有 PdfPrintable 而将其设为抽象），并且它按预期工作。

Answer 2

通过 default= 执行此操作确实感觉更好，但是您在那里使用的任何东西都无法获得您的班级或特定模型。要让它在管理员等地方正确显示，您可以在 init() 中设置它而不是 save()。

class OrderDocumentBase(PdfPrintable):
    number = models.PositiveIntegerField()

    def __init__(self, *args, **kwargs):
        super(OrderDocumentBase, self).__init__(*args, **kwargs)
        if not self.pk and not self.number:
            self.number = self.DEFAULT_NUMBER

class Invoice(OrderDocumentBase):
    DEFAULT_NUMBER = 2

class CreditAdvice(OrderDocumentBase):
    DEFAULT_NUMBER = 3