Handler if product_slug is given (wip)

This commit is contained in:
PCoder 2018-10-02 10:02:02 +02:00
parent 3bad37c605
commit e47f4f05b4

View file

@ -6,7 +6,7 @@ from django.contrib import messages
from django.contrib.auth import login, authenticate from django.contrib.auth import login, authenticate
from django.core.exceptions import ValidationError from django.core.exceptions import ValidationError
from django.core.urlresolvers import reverse from django.core.urlresolvers import reverse
from django.http import HttpResponseRedirect, JsonResponse from django.http import HttpResponseRedirect, JsonResponse, Http404
from django.shortcuts import render from django.shortcuts import render
from django.utils.translation import get_language, ugettext_lazy as _ from django.utils.translation import get_language, ugettext_lazy as _
from django.views.decorators.cache import cache_control from django.views.decorators.cache import cache_control
@ -254,6 +254,9 @@ class PaymentOrderView(FormView):
context.update({'generic_payment_form': GenericPaymentForm( context.update({'generic_payment_form': GenericPaymentForm(
prefix='generic_payment_form' prefix='generic_payment_form'
), }) ), })
# TODO: handle if we have a product id
#if 'product_id' in self.request.session:
else: else:
context.update({ context.update({
'vm_pricing': VMPricing.get_vm_pricing_by_name( 'vm_pricing': VMPricing.get_vm_pricing_by_name(
@ -269,6 +272,19 @@ class PaymentOrderView(FormView):
request.session['generic_payment_type'] = request.GET['type'] request.session['generic_payment_type'] = request.GET['type']
if 'generic_payment_details' in request.session: if 'generic_payment_details' in request.session:
request.session.pop('generic_payment_details') request.session.pop('generic_payment_details')
if 'product_slug' in kwargs:
logger.debug("Product slug is " + kwargs['product_slug'])
try:
product = GenericProduct.objects.get(
product_slug=kwargs['product_slug']
)
except GenericProduct.DoesNotExist as dne:
logger.error(
"Product '{}' does "
"not exist".format(kwargs['product_slug'])
)
raise Http404()
request.session['product_id'] = product.id
elif 'specs' not in request.session: elif 'specs' not in request.session:
return HttpResponseRedirect(reverse('datacenterlight:index')) return HttpResponseRedirect(reverse('datacenterlight:index'))
return self.render_to_response(self.get_context_data()) return self.render_to_response(self.get_context_data())