def setUp(self):
self.user_form = UserForm(self.data)
self.user_form.is_valid()
self.member_form = MemberForm(self.data)
self.member_form.is_valid()
self.user_instance = self.user_form.save()
self.member_instance = self.member_form.save(self.user_instance)
def add_member():
form = MemberForm()
if form.validate_on_submit():
data = Members(name=form.member_name.data, contact=form.contact_no.data, email=form.email.data,
dob=form.dob.data)
db.session.add(data)
db.session.commit()
flash("Member added successfully")
return render_template("add_member.html", form=form)
def edit_member(id):
form = MemberForm()
member = Members.query.get_or_404(id)
if form.validate_on_submit():
Members.query.filter_by(id=id).update(
{Members.name: form.member_name.data, Members.contact: form.contact_no.data,
Members.email: form.email.data, Members.dob: form.dob.data})
db.session.commit()
flash("member details edited successfully")
return redirect(url_for('members_bp.edit_member', id=member.id))
form.member_name.default = member.name
form.contact_no.default = member.contact
form.email.default = member.email
form.dob.default = member.dob
form.process()
return render_template("edit_member.html", form=form, member=member)
def test_change_category_with_pending_payments(self):
payment = Payment.objects.create(member=self.member_instance,
type=PaymentType.objects.get(id=1))
transaction = Transaction.objects.create(payment=payment,
status=0,
code='fakecode',
price='0.0')
self.member_instance.save()
member_form = MemberForm(instance=self.member_instance,
data={'category': 2})
self.assertFalse(self.member_instance.get_payment_status())
self.assertFalse(member_form.is_valid())
self.assertEqual(
member_form.errors['category'],
[u"You can't change your category with pending payments"])
def member_form(request):
try:
member = Member.objects.get(user=request.user)
except Member.DoesNotExist:
member = Member()
user_form = UserForm(request.POST or None, instance=request.user)
member_form = MemberForm(request.POST or None, instance=member)
if request.POST:
if member_form.is_valid() and user_form.is_valid():
member_form.save(user=request.user)
user_form.save()
messages.add_message(request, messages.INFO, _('Your data was updated successfully'))
return HttpResponseRedirect(reverse('members-dashboard'))
else:
messages.add_message(request, messages.ERROR, _('An error occurred while trying to save your data. check the form below. '))
return render(
request,
"members/member_form.html", {
"member_form": member_form,
'user_form': user_form
}
)
def setUp(self):
self.member_form = MemberForm(self.data)
def setUp(self):
self.member_form = MemberForm({})
