Django M2M форма Сохранить «через» таблицу
Вопрос
У меня проблемы с сохранением данных M2M, содержащие таблицу классов «через». Я хочу сохранить все выбранные элементы (выбранные в форме) во время через таблицу. Но я не знаю, как инициализировать таблицу «через» в поле зрения.
Мой код:
class Classroom(models.Model):
user = models.ForeignKey(User, related_name = 'classroom_creator')
classname = models.CharField(max_length=140, unique = True)
date = models.DateTimeField(auto_now=True)
open_class = models.BooleanField(default=True)
members = models.ManyToManyField(User,related_name="list of invited members", through = 'Membership')
class Membership(models.Model):
accept = models.BooleanField(User)
date = models.DateTimeField(auto_now = True)
classroom = models.ForeignKey(Classroom, related_name = 'classroom_membership')
member = models.ForeignKey(User, related_name = 'user_membership')
И по мнению:
def save_classroom(request):
classroom_instance = Classroom()
if request.method == 'POST':
form = ClassroomForm(request.POST, request.FILES, user = request.user)
if form.is_valid():
new_obj = form.save(commit=False)
new_obj.user = request.user
new_obj.save()
membership = Membership(member = HERE SELECTED ITEMS FROM FORM,classroom=new_obj)
membership.save()
Как я должен инициализировать членство на таблицу членства, чтобы быть заполненным правым?
Решение
В случае использования нормального соотношения M2M (не через посредническую таблицу) вы можете заменить:
membership = Membership(member = HERE SELECTED ITEMS FROM FORM,classroom=new_obj)
membership.save()
с
form.save_m2m()
Но в случае использования промежуточных таблиц вам необходимо вручную обрабатывать данные пост и создавать объекты членства со всеми обязательными полями (Аналогичная проблема). Наиболее основным решением является изменение вашего представления на что-то вроде:
def save_classroom(request):
if request.method == 'POST':
form = ClassroomForm(request.POST, request.FILES)
if form.is_valid():
new_obj = form.save(commit=False)
new_obj.user = request.user
new_obj.save()
for member_id in request.POST.getlist('members'):
membership = Membership.objects.create(member_id = int(member_id), classroom = new_obj)
return HttpResponseRedirect('/')
else:
form = ClassroomForm()
return render_to_response('save_classroom.html', locals())
Обратите внимание, как request.post манипулируется (.getlist). Это потому что пост и получить Напрокат Объекты, которые имеют некоторые последствия (request.post [«члены»], всегда вернутся один объект!).
Вы можете изменить этот код, чтобы получить его более надежным (обращение с ошибкой и т. Д.), И более Verbose, например:
member = get_object_or_404(User, pk = member_id)
membership = Membership.objects.create(member = member , classroom = new_obj)
Но обратите внимание, что вы выполняете некоторые запросы БД в цикле, которая не является хорошей идеей в целом (с точки зрения производительности).
Другие советы
Как и сделал то, что сделал Дзида, но использовать Form.Cleaned_data вместо запроса .post:
def save_classroom(request):
if request.method == 'POST':
form = ClassroomForm(request.POST, request.FILES)
if form.is_valid():
new_obj = form.save(commit=False)
new_obj.user = request.user
new_obj.save()
for member in form.cleaned_data['members'].all():
Membership.objects.create(member = member, classroom = new_obj)
return HttpResponseRedirect('/')
else:
form = ClassroomForm()
return render_to_response('save_classroom.html', locals())
Вам также необходимо рассмотреть некоторые членства могут быть удалены, поэтому:
def save_classroom(request):
if request.method == 'POST':
form = ClassroomForm(request.POST, request.FILES)
if form.is_valid():
new_obj = form.save(commit=False)
new_obj.user = request.user
new_obj.save()
final_members = form.cleaned_data['members'].all()
initial_members = form.initial['members'].all()
# create and save new members
for member in final_members:
if member not in initial_members:
Membership.objects.create(member = member, classroom = new_obj)
# delete old members that were removed from the form
for member in initial_members:
if member not in final_members:
Membership.objects.filter(member = member, classroom = new_obj).delete()
return HttpResponseRedirect('/')
else:
form = ClassroomForm()
return render_to_response('save_classroom.html', locals())
Если вы используете модели модели (например, в общем CBV: form_class=ClassroomForm
), переопределите и поместите сохраненную логику выше в save
Метод, что-то вроде:
ClassroomForm(forms.ModelForm):
members = ModelMultipleChoiceField(
queryset=Classroom.objects.all(),
widget=SelectMultiple
)
def save(self, commit=True):
classroom = super().save(commit=False)
if commit:
classroom.save()
if 'members' in self.changed_data:
final_members = self.cleaned_data['members'].all()
initial_members = self.initial['members']
# create and save new members
for member in final_members:
if member not in initial_members:
Membership.objects.create(member = member, classroom = new_obj)
# delete old members that were removed from the form
for member in initial_members:
if member not in final_members:
Membership.objects.filter(member = member, classroom = new_obj).delete()
return classroom
Вам также необходимо указать классную комнату для членства:
membership = Membership(member = request.user,
classroom=new_obj) #if new_obj if your classroom
membership.save()
Я думаю, вы также должны удалить User
в accept = models.BooleanField(User)
. Отказ Не должно быть необходимо установить дату после сохранения, если вы используете auto_now
Действительно Но, возможно, auto_now_add, скорее всего, вам нужно (http://docs.djangoproject.com/en/dev/ref/models/fields/#django.db.models.datefield.)
Вот как я сделал это в общем виде на основе класса обновления (Django 1.8) для аналогичного, но другого приложения, используя метод form_valid.
def form_valid(self, form):
"""
If the form is valid, save the associated model.
"""
self.object.members.clear()
self.object = form.save(commit=False)
self.object.user = self.request.user
self.object.save()
list_of_members = form.cleaned_data['members']
ClassRoom.objects.bulk_create([
Membership(
Course=self.object,
member=member_person,
order=num)
for num, member_person in enumerate(list_of_members)
])
return super(ModelFormMixin, self).form_valid(form)