django模型，如何做两个可选字段，但是一个填的时候，另一个也要填

Question

所以，我创建了一个具有一些属性的模型，但我想专注于这两个。

class Profile(models.Model):
    RANK_OPTIONS = (
        ('A', 'A'),
        ('B', 'B'),
    )
    related_office = models.ForeignKey('OtherModule.office', 
        related_name='office',
        on_delete=models.CASCADE
    )
    rank = models.CharField(('rank'),max_length=1, choices=RANK_OPTIONS)

基本上，我需要这两个属性是可选的，但是当属性“Rank”被填充时，属性related_office也必须被填充。 你可以有一个related_office 而不是一个等级，但是一旦你有了等级，你也需要一个相关的办公室。

我该怎么做？

Answer 1

您可以通过覆盖save 方法来强制执行此类条件：

def save(self, *args, **kwargs):
    if self.rank and not self.related_office:
        raise ValueError("related_office is required!")
    super().save(*args, **kwargs)

Answer 2

我不会在管理员保存方法中这样做。如果它在您的管理员中，请编写一个 ModelForm 并将其传递给您的管理员类，如下所示：

class ProfileForm(forms.ModelForm):
    class Meta:
        model = Profile
        fields = "__all__"

    def clean(self):
        cleaned_data = super().clean()
        if cleaned_data.get("rank") and not cleaned_data.get("related_office"):
            raise ValidationError('You have to select your related office for your rank')

        return cleaned_data

并在您的管理类中引用该表单：

class ProfileAdmin(admin.ModelAdmin):
    form = ProfileForm

如果您有自定义 html 模板，请在前端或后端对其进行验证。就像上面的例子一样

Answer 3

您可以在您的 model.py 中覆盖 clean 方法：

def clean(self):
    if self.rank and not self.related_office:
        raise forms.ValidationError({'related_office':["related_office is required!"]})