def test_form(self):
form_class = email_invite_form_factory(self.user)
data = {'email': '*****@*****.**'}
form = form_class(data)
self.assertTrue(form.is_valid())
inv = form.save()
self.assertIsInstance(inv, EmailInvite)
self.assertEqual(inv.user, self.user)
self.assertEqual(inv.email, data['email'])
def email_invites(request):
form_class = email_invite_form_factory(request.user)
formset_class = modelformset_factory(EmailInvite, form=form_class, extra=4, formset=BaseEmailInviteFormSet)
personal_note_form = PersonalNoteForm(request.POST or None)
formset = formset_class(request.POST or None)
if formset.is_valid():
invites = formset.save(commit=False)
if personal_note_form.is_valid():
note = personal_note_form.cleaned_data['personal_note']
for i in invites:
i.personal_note = note
i.save()
messages.success(request, _('Invites sent successfully!'))
return redirect("home")
return direct_to_template(request, 'invite_friends/email_invite.html', locals())
def email_invites(request):
form_class = email_invite_form_factory(request.user)
formset_class = modelformset_factory(EmailInvite,
form=form_class,
extra=4,
formset=BaseEmailInviteFormSet)
personal_note_form = PersonalNoteForm(request.POST or None)
formset = formset_class(request.POST or None)
if formset.is_valid():
invites = formset.save(commit=False)
if personal_note_form.is_valid():
note = personal_note_form.cleaned_data['personal_note']
for i in invites:
i.personal_note = note
i.save()
messages.success(request, _('Invites sent successfully!'))
return redirect("home")
return direct_to_template(request, 'invite_friends/email_invite.html',
locals())
