python-3.x - 如何告诉 Django 模型只有一个或其他字段必须不为空？

Question

使用 Django models.Model 类:

class OrderLign(models.Model):
    Order = models.ForeignKey(Order,on_delete=models.CASCADE)
    Item = models.ForeignKey(Item, on_delete=models.RESTRICT, blank=True, default=0, null=True)
    Pack = models.ForeignKey(Pack,on_delete=models.RESTRICT, blank=True, default=0,
                             null=True)

我需要只有在设置了 Item OR Pack 时才可以保存 OrderLign，不能同时设置，也不能两者都不设置。

Answer 1

您可以添加覆盖 clean() method ^[Django-doc]检查这个。

此外，我们可以使用 Django's constraint API 添加数据库约束并使用 CheckConstraint ^[Django-doc] .如果数据库支持这一点，它将拒绝两个 ForeignKey 都没有的条目，或者如果两个都不是 None:

from django.core.exceptions import ValidationError
from django.db.models import Q

class OrderLign(models.Model):
    order = models.ForeignKey(
        Order,
        on_delete=models.CASCADE
    )
    item = models.ForeignKey(
        Item,
        on_delete=models.RESTRICT,
        blank=True,
        null=True
    )
    pack = models.ForeignKey(
        Pack,
        on_delete=models.RESTRICT,
        blank=True,
        null=True
    )

    def clean(self, *args, **kwargs):
        if (self.item is None) == (self.pack is None):
            raise ValidationError('item or pack should be filled in')
        return super().clean(*args, **kwargs)

    class Meta:
        constraints = [
            models.CheckConstraint(
                check=Q(item=None, pack__isnull=False) |
                      Q(pack=None, item__isnull=False),
                name='one of the two not null'
            )
        ]