Django ManyToMany through表单如何设置ModelChoiceField过滤queryset

你之前的实现冲突核心是两点：

• GET请求渲染表单时手动给字段赋值了queryset，但POST请求重新实例化ProgramExamForm时没有做同样的赋值，此时exam字段的queryset还是初始的空集合，提交的选项会被判定为非法值
• 把queryset过滤逻辑散落在视图层，容易因为实例化场景不同漏写配置，维护成本高

1. 封装表单过滤逻辑 #

把queryset过滤逻辑收敛到表单类的初始化方法里，通过额外参数传入所属学校实例，不管是GET还是POST场景实例化表单，只要传入对应参数就能自动完成过滤，不会出现逻辑遗漏。

from django import forms
from .models import ProgramExam, Exam

class ProgramExamForm(forms.ModelForm):
    exam = forms.ModelChoiceField(queryset=Exam.objects.none())

    class Meta:
        model = ProgramExam
        fields = ['exam', 'coef']

    def __init__(self, *args, **kwargs):
        # 提取自定义传入的university参数，避免传给父类触发参数错误
        university = kwargs.pop('university', None)
        super().__init__(*args, **kwargs)
        if university:
            self.fields['exam'].queryset = Exam.objects.filter(university=university)

2. 修正视图层实例化逻辑 #

不管是处理GET空表单渲染，还是POST提交的表单校验，实例化ProgramExamForm时都传入当前用户关联的学校实例，同时修正原有代码里的变量名笔误（原代码里误写了program_form，实际变量名是program_exam_form）。

from django.http import HttpResponseForbidden

def create_program(request):
    user = request.user
    # 前置权限校验
    if not (user.is_authenticated and hasattr(user, 'university')):
        return HttpResponseForbidden("无操作权限")
    
    target_university = user.university
    if request.method == 'POST':
        # POST场景实例化表单必须传入university参数，保证校验时queryset正确
        program_exam_form = ProgramExamForm(request.POST, university=target_university)
        if program_exam_form.is_valid():
            program_exam = program_exam_form.save(commit=False)
            # 注意：如果是新建Program的流程，需要在这里给program_exam绑定对应的Program实例后再保存
            # program_exam.program = 新创建的Program对象
            program_exam.save()
            # 补充你的成功跳转逻辑
    else:
        # GET场景初始化空表单
        program_exam_form = ProgramExamForm(university=target_university)

    context = {'program_exam_form': program_exam_form}
    return render(request, 'create_program.html', context)

注意事项 #

• 如果是编辑已有Program的场景，直接传入当前Program关联的university实例即可，不需要从当前用户取，能自动避免跨校越权提交的问题
• 由于ProgramExam中Exam外键配置的是on_delete=models.DO_NOTHING，删除Exam记录时要提前处理关联的中间表数据，避免产生无效关联
• 若表单提交时提示exam选项非法，优先检查实例化表单时传入的university是否正确，对应queryset是否包含提交的Exam实例

内容的提问来源于stack exchange，提问作者PGMatev