From f502e53845b32f7807ed3d71a52c45b826074272 Mon Sep 17 00:00:00 2001
From: PCoder <purple.coder@yahoo.co.uk>
Date: Mon, 1 Jul 2019 06:45:48 +0530
Subject: [PATCH] Add basic implementation of AskSSHKeyView

---
 datacenterlight/views.py                | 22 +++++++++++++++++++++-
 hosting/templates/hosting/user_key.html |  3 ++-
 2 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/datacenterlight/views.py b/datacenterlight/views.py
index 76f50aec..d76303f7 100644
--- a/datacenterlight/views.py
+++ b/datacenterlight/views.py
@@ -523,13 +523,33 @@ class PaymentOrderView(FormView):
             else:
                 request.session['customer'] = customer
             return HttpResponseRedirect(
-                reverse('datacenterlight:order_confirmation'))
+                reverse('datacenterlight:add_ssh_key'))
         else:
             context = self.get_context_data()
             context['billing_address_form'] = address_form
             return self.render_to_response(context)
 
 
+class AskSSHKeyView(FormView):
+    form_class = UserHostingKeyForm
+    template_name = "datacenterlight/add_ssh_key.html"
+
+    def get_form_kwargs(self):
+        kwargs = super(AskSSHKeyView, self).get_form_kwargs()
+        kwargs.update({'request': self.request})
+        return kwargs
+
+    @cache_control(no_cache=True, must_revalidate=True, no_store=True)
+    def get(self, request, *args, **kwargs):
+        context = {
+            'site_url': reverse('datacenterlight:index'),
+            'cms_integration': get_cms_integration('default'),
+            'form': UserHostingKeyForm(request=self.request),
+            'keys': get_all_public_keys(self.request.user)
+        }
+        return render(request, self.template_name, context)
+
+
 class OrderConfirmationView(DetailView, FormView):
     form_class = UserHostingKeyForm
     template_name = "datacenterlight/order_detail.html"
diff --git a/hosting/templates/hosting/user_key.html b/hosting/templates/hosting/user_key.html
index 804d661a..247551b5 100644
--- a/hosting/templates/hosting/user_key.html
+++ b/hosting/templates/hosting/user_key.html
@@ -8,7 +8,8 @@
                 <form method="POST" action="" novalidate class="form-ssh">
                     {% csrf_token %}
                     <div class="page-header">
-                        <h1 class="h1-thin"><i class="fa fa-key" aria-hidden="true"></i>&nbsp;{% trans "Add your public SSH key" %}</h1>
+                        <h1 class="h1-thin"><i class="fa fa-key" aria-hidden="true"></i>&nbsp;{% if title %}{% trans title %}{% else %} {% endif %}</h1>
+                        {% if sub_title %}<span>{% trans sub_title %}</span>{% else %}{% endif %}
                     </div>
                     {% if messages %}
                         <div class="alert alert-warning">