From f3b61db782ba6548a7e34dd8b7b6e47933b8080d Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Mon, 26 Feb 2018 21:20:08 +0300
Subject: [PATCH] LIL-268, LIL-269. Hide payment button

---
 apps/course/templates/course/course.html | 16 +++++++++-------
 apps/course/views.py                     | 10 ++++++++++
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/apps/course/templates/course/course.html b/apps/course/templates/course/course.html
index 6bd03bbc..fdfac3d8 100644
--- a/apps/course/templates/course/course.html
+++ b/apps/course/templates/course/course.html
@@ -25,14 +25,16 @@
         </div>
         <div class="go__title">Вернуться</div>
       </a>
-      <button 
+      {% if not paid or not pending %}
+      <a
         class="go__btn btn btn_md"
         {% if user.is_authenticated %}
-        data-popup=".js-popup-buy"
+        href="{% url 'course-checkout' course.id %}"
         {% else %}
         data-popup=".js-popup-auth"
         {% endif %}
-        >КУПИТЬ КУРС</button>
+        >КУПИТЬ КУРС</a>
+      {% endif %}
     </div>
     <div
       class="course"
@@ -156,7 +158,7 @@
         <img class="video__pic" src="{{ course.cover.image.url }}"/>
         {% else %}
         <img class="video__pic" src="{% static 'img/no_cover.png' %}"/>
-        {% endif %}      
+        {% endif %}
         {% if course.is_deferred_start %}
         <div class="video__soon">
           <div class="video__title">Курс начнется:</div>
@@ -306,7 +308,7 @@
   </div>
 </div>
 {% endif %}
-<div 
+<div
   class="course course_promo"
   style="background-image: url({% if course.cover %}{{ course.cover.image.url }}{% else %}{% static 'img/no_cover.png' %}{% endif %});"
   >
@@ -383,7 +385,7 @@
           <div class="meta__title">{{ course.price|floatformat:"-2" }}₽</div>
         </div>
       </div>
-      <button 
+      <button
         class="course__buy btn btn_md"
         {% if user.is_authenticated %}
         data-popup=".js-popup-buy"
@@ -459,4 +461,4 @@
     </div>
   </div>
 </div>
-{% endblock content %}
\ No newline at end of file
+{% endblock content %}
diff --git a/apps/course/views.py b/apps/course/views.py
index 01c3ea95..98fc5a4f 100644
--- a/apps/course/views.py
+++ b/apps/course/views.py
@@ -8,6 +8,8 @@ from django.views.generic import View, CreateView, DetailView, ListView, Templat
 from django.utils.decorators import method_decorator
 from django.views.decorators.csrf import csrf_exempt
 from django.views.decorators.http import require_http_methods
+
+from apps.payment.models import AuthorBalance
 from .models import Course, Like, Lesson, CourseComment, LessonComment
 from .filters import CourseFilter
 
@@ -187,6 +189,14 @@ class CourseView(DetailView):
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
         context['next'] = self.request.GET.get('next', None)
+        context['paid'] = self.object.payments.filter(
+            user=self.request.user,
+            authorbalance__status=AuthorBalance.ACCEPTED,
+        ).exists()
+        context['pending'] = self.object.payments.filter(
+            user=self.request.user,
+            authorbalance__status=AuthorBalance.PENDING,
+        ).exists()
         return context
 
     def get_queryset(self):