From ee80181a5507cbdc848872efc72f37a8f672bc13 Mon Sep 17 00:00:00 2001
From: Levi <levinoelvm@gmail.com>
Date: Wed, 24 May 2017 10:58:16 -0500
Subject: [PATCH] added redirect after create vm ssh key

---
 datacenterlight/views.py                      |  3 ++
 .../hosting/virtual_machine_key.html          |  2 +-
 hosting/views.py                              | 29 +++++++++++++------
 3 files changed, 24 insertions(+), 10 deletions(-)

diff --git a/datacenterlight/views.py b/datacenterlight/views.py
index 3e6a7141..b68f7408 100644
--- a/datacenterlight/views.py
+++ b/datacenterlight/views.py
@@ -51,6 +51,9 @@ class PricingView(TemplateView):
 
         request.session['template'] = VirtualMachineTemplateSerializer(template).data
 
+        if not request.user.is_authenticated():
+            request.session['next'] = reverse('hosting:payment')
+
         request.session['specs'] = { 
             'cpu':cores,
             'memory': memory,
diff --git a/hosting/templates/hosting/virtual_machine_key.html b/hosting/templates/hosting/virtual_machine_key.html
index 157ceea8..f7780466 100644
--- a/hosting/templates/hosting/virtual_machine_key.html
+++ b/hosting/templates/hosting/virtual_machine_key.html
@@ -112,7 +112,7 @@
 </script>
 {%endif%}
 
-
+{{next_url}}
 {% if next_url %}
     <script type="text/javascript"> 
         window.location.href = '{{next_url}}';
diff --git a/hosting/views.py b/hosting/views.py
index b8b04b7d..3999037b 100644
--- a/hosting/views.py
+++ b/hosting/views.py
@@ -319,12 +319,22 @@ class GenerateVMSSHKeysView(LoginRequiredMixin, FormView):
         form.save()
         context = self.get_context_data()
 
+        next_url = self.request.session.get(
+            'next',
+            reverse('hosting:create_virtual_machine')
+        )
+
+        if 'next' in self.request.session:
+            context.update({
+                'next_url': next_url
+            })
+            del (self.request.session['next'])
+
         if form.cleaned_data.get('private_key'):
             context.update({
                 'private_key': form.cleaned_data.get('private_key'),
                 'key_name': form.cleaned_data.get('name'),
                 'form': UserHostingKeyForm(request=self.request),
-                'next_url': reverse('hosting:create_virtual_machine')
             })
 
         # return HttpResponseRedirect(reverse('hosting:key_pair'))
@@ -391,9 +401,6 @@ class PaymentVMView(LoginRequiredMixin, FormView):
         return context
 
     def get(self, request, *args, **kwargs):
-        if 'next' in request.session:
-            del request.session['next']
-
         try:
             UserHostingKey.objects.get(
                 user=self.request.user
@@ -405,6 +412,9 @@ class PaymentVMView(LoginRequiredMixin, FormView):
             )
             return HttpResponseRedirect(reverse('hosting:key_pair'))
 
+        if 'next' in request.session:
+            del request.session['next']
+
         return self.render_to_response(self.get_context_data())
 
     def post(self, request, *args, **kwargs):
@@ -628,16 +638,17 @@ class CreateVirtualMachinesView(LoginRequiredMixin, View):
 
             context = {
                 'templates': VirtualMachineTemplateSerializer(templates, many=True).data,
-                'configuration_options' : configuration_options,
+                'configuration_options': configuration_options,
             }
         except:
-            messages.error( request,
+            messages.error(
+                request,
                 'We could not load the VM templates due to a backend connection \
                 error. Please try again in a few minutes'
-                )
+            )
             context = {
-                'error' : 'connection'
-                    }
+                'error': 'connection'
+            }
 
         return render(request, self.template_name, context)