只允许在 Django 模型和管理员中填写一个或另一个字段

Question

我有一个用于 Group 的简单 Django 模型，其中包含 Contacts 列表。每个组必须选择任一主要联系人 ForeignKey 或所有联系人 BooleanField，但不能同时选择两者，也不能没有。

class Group(models.Model):
    contacts = models.ManyToManyField(Contact)
    contact_primary = models.ForeignKey(Contact, related_name="contact_primary", null=True)
    all_contacts = models.BooleanField(null=True)

我如何确保：

1. 除非设置了contact_primary 或all_contacts（但不能同时设置），否则无法保存模型。我想那是通过实现Group.save() 方法？还是应该是Group.clean() 方法？？

2. 在 Django Admin 中，要么在选择一个字段时禁用另一个字段，要么至少提供一个有意义的错误消息，如果两者都设置或没有设置并且管理员尝试保存它？

谢谢！

Answer 1

最简单的方法是覆盖模型的save() 方法：

class Group(models.Model):
    contacts = models.ManyToManyField(Contact)
    contact_primary = models.ForeignKey(Contact, related_name="contact_primary", blank=True, null=True)
    all_contacts = models.BooleanField(blank=True, null=True)

    def save(self, *args, **kwargs):
        if self.contact_primary is not None and self.all_contacts is not None:
            raise Exception("some error message here")  
        if self.contact_primary is None and self.all_contacts is None:
            raise Exception("some other error message here")  

        return super().save()

请注意，我在您的模型字段中添加了 blank=True。如果您想通过管理员或表单插入空列，这是必要的。

更新

如果您想提升 ValidationError，您必须从模型的 clean() 方法提升它。否则，您将给客户端一个 500 错误，而不是错误消息。