Created current user billing address model in membership payment

This commit is contained in:
Levi 2016-10-13 23:50:13 -05:00
parent 082f39e9dc
commit eaf1f1de86

View file

@ -366,6 +366,15 @@ class MembershipPaymentView(LoginRequiredMixin, IsNotMemberMixin, FormView):
# Create Billing Address
billing_address = form.save()
# Create Billing Address for User if he does not have one
if not customer.user.billing_addresses.count():
data.update({
'user': customer.user.id
})
billing_address_user_form = UserBillingAddressForm(data)
billing_address_user_form.is_valid()
billing_address_user_form.save()
# Create membership plan
membership_data = {'type': membership_type}
membership = Membership.create(membership_data)
@ -460,6 +469,10 @@ class UserBillingAddressView(LoginRequiredMixin, UpdateView):
def get_form_kwargs(self):
current_billing_address = self.request.user.billing_addresses.first()
form_kwargs = super(UserBillingAddressView, self).get_form_kwargs()
if not current_billing_address:
return form_kwargs
form_kwargs.update({
'initial': {
'street_address': current_billing_address.street_address,