From 165ac380354854a44c5641a265ea0527adb5895a Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Wed, 25 Apr 2018 17:20:03 +0300
Subject: [PATCH] Update school success payment template

---
 apps/payment/templates/payment/payment_success.html | 7 +++++++
 apps/payment/views.py                               | 3 +++
 2 files changed, 10 insertions(+)

diff --git a/apps/payment/templates/payment/payment_success.html b/apps/payment/templates/payment/payment_success.html
index 4d22d6fa..53d946d4 100644
--- a/apps/payment/templates/payment/payment_success.html
+++ b/apps/payment/templates/payment/payment_success.html
@@ -2,10 +2,17 @@
 <div class="section">
   <div class="section__center center center_xs">
     <div class="done">
+      {% if school %}
+      <div class="done__title title">Вы успешно приобрели школу!</div>
+      <div class="done__foot">
+        <a class="done__btn btn btn_md btn_stroke" href="{% url 'school:school' %}">ПЕРЕЙТИ К ШКОЛЕ</a>
+      </div>
+      {% else %}
       <div class="done__title title">Вы успешно приобрели курс!</div>
       <div class="done__foot">
         <a class="done__btn btn btn_md btn_stroke" href="/">ПЕРЕЙТИ К ГЛАВНОЙ</a>
       </div>
+      {% endif %}
     </div>
   </div>
 </div>
diff --git a/apps/payment/views.py b/apps/payment/views.py
index 90e3a560..bb44f02f 100644
--- a/apps/payment/views.py
+++ b/apps/payment/views.py
@@ -38,6 +38,9 @@ class CourseBuySuccessView(TemplateView):
 class SchoolBuySuccessView(TemplateView):
     template_name = 'payment/payment_success.html'
 
+    def get(self, request, pk=None, *args, **kwargs):
+        return self.render_to_response(context={'school': True})
+
 
 @method_decorator(login_required, name='dispatch')
 class CourseBuyView(TemplateView):