def step_one(request, code):
request.step_num = 1
user = request.user
form = SampleForm(request.POST)
if request.method == 'POST' and form.validate():
#Store main.
sample = Sample.objects.create(
user_ref=user.user_ref,
sample_data=form.data
)
messages.add_message(request, messages.SUCCESS, 'Sample Created and Added to Cart')
if form.data.get('country') == 'US':
return redirect(reverse('step_one_validate', kwargs={'pending_id': sample.pk}))
return redirect(reverse('step_two'))
return {'form': form}
def step_one_edit(request, pending_id):
request.step_num = 1
form = SampleForm(request.POST)
try:
pending = Sample.objects.get(pk=pending_id)
except Sample.DoesNotExist:
messages.add_message(request, messages.ERROR, 'Sample Not Available to Edit')
return redirect(reverse('step_two'))
if request.method == 'POST' and form.validate():
pending.sample_data = form.data
pending.save()
if form.data.get('country') == 'US':
return redirect(reverse('step_one_validate', kwargs={'pending_id': pending_id}))
messages.add_message(request, messages.SUCCESS, 'Sample Saved')
return redirect(reverse('step_two'))
else:
form = SampleForm(**pending.sample_data) #Since this is a dictionary, not an object... KW population.
return {'form': form, 'obj': pending}
