Django поле множественного выбора, form.save() перезаписывает все значения, сбрасывая предыдущие записи, а не добавляя новые
У меня есть форма, предназначенная для добавления нескольких разделов в проект. Я использую CheckboxSelectMultiple со списком вариантов, ограниченным только теми, которые не были выбраны ранее.
Итак, с новым пустым проектом без разделов:
Project.sections.all() = null
choices = [(1, 'a'),(2, 'b'),(3, 'c')]
При первом отправлении формы добавляются секции a & b.
Project.sections.all() = a, b
choices = [(3, 'c')]
Второй раз, когда форма отправляется, добавляется секция c.
Project.sections.all() = c
choices = [(1, 'a'),(2, 'b')]
Вместо этого я хочу, чтобы c было добавлено в существующий список значений для проекта.
models.py
class Project(models.Model):
number = models.CharField(max_length=4, unique=True)
name = models.CharField(max_length=100)
sections = models.ManyToManyField(Section)
class Section(models.Model):
code = models.CharField(max_length=2)
description = models.CharField(max_length=50)
views.py
def add_section(request, project_number):
project = Project.objects.get(number=project_number)
full_section_list = Section.objects.all()
project_assigned_sections = project.sections.all().values_list('id', flat=True)
choices = list(full_section_list.exclude(pk__in=project_assigned_sections).values_list('id', 'code'))
if request.method == 'POST':
form = AddSectionForm(choices, request.POST, instance=project)
if form.is_valid():
form.save()
return HttpResponseRedirect(reverse("project-page", args=(project_number)))
else:
print("invalid")
else:
form = AddSectionForm(choices, instance=project)
return render(request, "app/add_section.html", {
"project": project,
"form": form,
"choices": choices
})
forms.py
class AddSectionForm(forms.ModelForm):
def __init__(self, choices, *args, **kwargs):
super(AddSectionForm, self).__init__(*args, **kwargs)
self.fields['sections'] = forms.MultipleChoiceField(
widget=forms.CheckboxSelectMultiple,
required=False,
choices=choices
)
class Meta:
model = Project
fields = ['sections']
Я предполагаю, что Django делает то, для чего он предназначен... так что, работайте вокруг этого.
if request.method == 'POST':
previous_sections = list( project.sections.values_list( 'pk', flat=True) )
form = AddSectionForm(choices, request.POST, instance=project)
if form.is_valid():
new = form.save( commit = False)
new.sections.add( previous_sections ) # make sure old ones are never removed
new.save()
return HttpResponseRedirect(reverse("project-page", args=(project_number)))
else:
print("invalid")
else:
form = AddSectionForm()
не передавать экземпляр, просто оставить пустую форму
У меня была некоторая удача с этим, а не с вызовом save(), хотя это не кажется таким уж приятным
data = form.cleaned_data.get("sections")
data_int = list(map(int,data))
project.sections.add(*data_int)
Ок, это работает и выглядит немного приятнее (я больше не отображаю целые числа в возвращаемый список).
data = form.cleaned_data.get("sections") #get the list of id's from the form
sections = Section.objects.filter(pk__in=data) #gets the objects from those id's
for section in sections: #iterate through them, adding to the project.
project.sections.add(section)