From af523690f0e06bc700397cdaa33ff22daad2fb8b Mon Sep 17 00:00:00 2001
From: Dmitriy Shesterkin <mitri4@bk.ru>
Date: Mon, 17 Jul 2017 19:59:46 +0300
Subject: [PATCH] success page

---
 src/robokassa/views.py           | 15 +++++++++++++--
 templates/robokassa/fail.html    |  2 +-
 templates/robokassa/success.html |  6 +++++-
 3 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/src/robokassa/views.py b/src/robokassa/views.py
index 45fa94e..c676b1c 100644
--- a/src/robokassa/views.py
+++ b/src/robokassa/views.py
@@ -4,7 +4,7 @@ from django.http import HttpResponse
 from django.template.response import TemplateResponse
 from django.views.decorators.csrf import csrf_exempt
 
-from customer.models import Payment
+from customer.models import Payment, License
 from customer.utils import get_robokassa_url
 from robokassa.conf import USE_POST
 from robokassa.forms import (
@@ -64,7 +64,18 @@ def success(request, template_name='robokassa/success.html',
             sender=form, InvId=inv_id, OutSum=out_sum,
             extra=form.extra_params())
 
-        context = {'InvId': inv_id, 'OutSum': out_sum, 'form': form}
+        try:
+            payment = Payment.objects.get(
+                pk=inv_id,
+                order_amount=out_sum
+            )
+            lic = License.objects.get(pk=payment.order_number)
+        except Payment.DoesNotExist:
+            lic = ''
+        except License.DoesNotExist:
+            lic = ''
+
+        context = {'InvId': inv_id, 'OutSum': out_sum, 'lic': lic, 'form': form}
         context.update(form.extra_params())
         context.update(extra_context or {})
         return TemplateResponse(request, template_name, context)
diff --git a/templates/robokassa/fail.html b/templates/robokassa/fail.html
index b869763..d3fdfb4 100644
--- a/templates/robokassa/fail.html
+++ b/templates/robokassa/fail.html
@@ -8,6 +8,6 @@
       <p>Что-то пошло не так и оплатить лицензию не получилось :(</p>
       <p>Пожалуйста, попробуйте снова по <a href="{{ pay_link }}">этой</a> ссылке</p>
       <p>Ссылку на оплату, историю всех расчётов и купленных лицензий всегда можно найти на странице
-        <a href="{% url 'customer-orders' %}">Лицензии и расчёты.</a></p>
+        <a href="{% url 'customer-orders' %}">Лицензии и расчёты</a></p>
     </div>
 {% endblock %}
diff --git a/templates/robokassa/success.html b/templates/robokassa/success.html
index 7b7c5eb..2bad108 100644
--- a/templates/robokassa/success.html
+++ b/templates/robokassa/success.html
@@ -5,7 +5,11 @@
 {% block right-column %}{% endblock %}
 {% block content %}
     <div id="content">
-      <h2>Спасибо!</h2>
+      <h2>Спасибо Вам за оплату счёта № {{ lic.pk }}.!</h2>
       <p>Ваш заказ принят!</p>
+
+<p>Для Вас была создана лицензия № {{ lic }} на {{ lic.get_term }}, которая будет автоматически активирована {{ lic.from_date|date 'd.m.Y' }}</p>
+<p>Акт выполненных работ будет создан в момент активации лицензии. Вы сможете скачать его на этой странице
+  <a href="{% url 'customer-orders' %}">Лицензии и расчёты</a></p>
     </div>
 {% endblock %}