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

Question

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

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

我查看了 这个 stackoverflow 问题，但它没有解决同样的问题。我认为唯一可行的是重载 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

通过 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

Answer 2

这里有几个问题。首先，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

我用一个小改动测试了上面的内容（使 OrderDocumentBase 抽象，因为我没有 PdfPrintable），并且它按预期工作。