def done(self, form_list, form_dict, **kwargs):
"""
Gather the data from the three forms. If the user already exists,
update the profile, if not create a new user. Then add a new membership.
"""
login_step = self.get_cleaned_data_for_step('login')
address_step = self.get_cleaned_data_for_step('user_info')['address']
profile_step = self.get_cleaned_data_for_step('user_info')['profile']
membership_step = self.get_cleaned_data_for_step('membership')
# Renew membership for current user
if login_step is None:
u = self.request.user
# Update the user profile
profile_form = ProfileForm(profile_step, instance=u.profile)
u.profile = profile_form.save()
u.save()
else:
# Create a new user from steps 'login' & 'profile'
u = User(email=login_step['email'])
u.is_active = False # Not activated yet
p = Profile(**profile_step)
p.save()
u.profile = p
u.save()
# Create or update address info
addresses = u.profile.address_set.all()
if len(addresses) > 0:
add = addresses[0]
add = AddressForm(address_step, instance=addresses[0])
else:
add = Address(**address_step)
add.profile = u.profile
add.save()
# Create a new Membership object
membership = Membership(**membership_step)
membership.profile = u.profile
membership.start_date = membership.next_start_date()
membership.amount = membership.compute_amount()
membership.duration = 1 # valid for one year
membership.save()
# Send a confirmation email if the user has to pay
if membership.amount > 0:
membership.email_user(status="submitted")
mb_url = self.request.build_absolute_uri(
reverse('membership-detail', args=[membership.uid]))
return redirect("membership-detail", membership.uid)
