CreateView不保存相关对象

Question

我有两个模型：学生模型和小组模型。 一组由多个学生组成，一个学生只能在一个组中。 这些是模型：

class Group(models.Model):
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

class Student(models.Model):
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    section = models.PositiveSmallIntegerField(default=1)
    group = models.ForeignKey(
        Group, on_delete=models.SET_NULL, null=True, blank=True
    )

我正在尝试建立一个表格，在这里我可以选择多个学生来创建一个小组。

class CreateGroupForm(forms.ModelForm):
    students = forms.ModelMultipleChoiceField(
        required=True,
        queryset=Student.objects.all()
    )

    class Meta:
        model = Group
        fields = ('students', )

我在表格中使用以下视图：

class SelectGroup(CreateView):
    model = Group
    form_class = CreateGroupForm
    template_name = 'selection.html'
    success_url = reverse_lazy('test')

当我提交表单时，它将创建一个组，但该组的student_set为空。 我猜这是因为如果不先保存组，就无法将学生添加到组中。 我是否可以修改此视图以保存学生，还是应该使用其他方法？

Answer 1

由于students不是小组模型的一个领域，因此模型表单的save不知道该如何处理。 您必须重写save方法并手动处理students域：

class CreateGroupForm(forms.ModelForm):
    # ...
    def save(self, commit=True):
        # you have to commit, else the reverse fk has nothing to point to
        group = super(CreateGroupForm, self).save(commit=True)
        group.student_set.add(*self.cleaned_data['students'])
        return group

如果您不想在表单上删除非提交保存选项，则还可以覆盖视图的form_valid方法：

class SelectGroup(CreateView):
    # ...
    def form_valid(self, form):
        self.object.student_set.add(*self.form.cleaned_data['students'])
        return super(SelectGroup, self).form_valid(form)