Django如何验证AnswerToQuestion多对多关联选项归属对应问题？

我定义了以下Django模型：

class Question(Model):
    # ...

class Choice(Model):
    question = ForeignKey(Question)
    # ...

class AnswerToQuestion(Model):
    user = ForeignKey(User)
    question = ForeignKey(Question)
    selected_choices = ManyToManyField(Choice)

我希望确保selected_choices中的选项始终属于question字段对应的问题。

我尝试在AnswerToQuestion的clean方法中进行验证：

def clean(self):
    super().clean()
    if self.pk and self.selected_choices.exclude(question=self.question).exists():
        raise ValidationError("Invalid choice(s)")

但该方法未达到预期效果：由于多对多关系在模型保存后才更新，添加无效选项时能成功保存，但后续每次保存都会失败，因为已存在无效关联选项。

请问还有其他实现该需求的方法吗？

解决方案 #

1. 用m2m_changed信号拦截无效关联 #

利用Django的m2m_changed信号，在多对多关系变更前验证每个待添加的选项：

from django.db.models.signals import m2m_changed
from django.dispatch import receiver
from django.core.exceptions import ValidationError

@receiver(m2m_changed, sender=AnswerToQuestion.selected_choices.through)
def validate_selected_choices(sender, instance, action, pk_set, **kwargs):
    if action == 'pre_add':
        # 检查要添加的Choice是否都属于当前Answer的question
        invalid_choices = Choice.objects.filter(pk__in=pk_set).exclude(question=instance.question)
        if invalid_choices.exists():
            raise ValidationError("选择的选项不属于当前问题")

这个信号会在多对多关系实际写入数据库前触发，直接阻止无效选项的关联，从源头避免错误数据。

2. 自定义多对多中间模型，添加数据库级约束 #

通过自定义中间模型，在数据库层面强制约束关联的Choice和Answer的Question一致：

class AnswerChoiceThrough(models.Model):
    answer = models.ForeignKey(AnswerToQuestion, on_delete=models.CASCADE)
    choice = models.ForeignKey(Choice, on_delete=models.CASCADE)

    class Meta:
        constraints = [
            models.CheckConstraint(
                check=models.Q(answer__question=models.F('choice__question')),
                name='choice_belongs_to_answer_question'
            )
        ]

class AnswerToQuestion(Model):
    user = ForeignKey(User)
    question = ForeignKey(Question)
    selected_choices = ManyToManyField(Choice, through=AnswerChoiceThrough)

这种方法能保证数据一致性，即使绕过ORM直接操作数据库也生效，但要注意：MySQL 8.0.16及以上版本才支持CheckConstraint，低版本需要用触发器替代。

3. 表单层面验证（适用于前端提交场景） #

如果通过表单创建或更新答案，可以在表单的clean方法中提前验证：

from django import forms

class AnswerToQuestionForm(forms.ModelForm):
    class Meta:
        model = AnswerToQuestion
        fields = ['question', 'selected_choices']

    def clean(self):
        cleaned_data = super().clean()
        question = cleaned_data.get('question')
        selected_choices = cleaned_data.get('selected_choices')
        
        if question and selected_choices:
            invalid_count = selected_choices.exclude(question=question).count()
            if invalid_count > 0:
                raise forms.ValidationError(f"有{invalid_count}个选项不属于当前问题")
        return cleaned_data

在数据进入模型前就完成验证，避免无效数据存入数据库。

4. 重写save_m2m方法（ModelForm场景） #

如果使用ModelForm保存，也可以重写save_m2m方法拦截错误：

class AnswerToQuestionForm(forms.ModelForm):
    # ... 元类定义

    def save_m2m(self):
        question = self.instance.question
        invalid_choices = self.cleaned_data['selected_choices'].exclude(question=question)
        if invalid_choices.exists():
            raise ValidationError("选择的选项不属于当前问题")
        super().save_m2m()

这种方式直接在保存多对多关系前做验证，和表单验证配合使用效果更可靠。

内容的提问来源于stack exchange，提问作者Samuele B.