From 1e5ac2516d0bcdaf0fd32f978f54f81ebd815b0a Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Mon, 5 Mar 2018 11:11:09 +0300
Subject: [PATCH] Fix check payment status

---
 apps/course/views.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/apps/course/views.py b/apps/course/views.py
index 98fc5a4f..7cdcaa8e 100644
--- a/apps/course/views.py
+++ b/apps/course/views.py
@@ -1,3 +1,5 @@
+from paymentwall import Pingback
+
 from django.contrib.auth import get_user_model
 from django.contrib.auth.decorators import login_required
 from django.db.models import Q
@@ -191,11 +193,15 @@ class CourseView(DetailView):
         context['next'] = self.request.GET.get('next', None)
         context['paid'] = self.object.payments.filter(
             user=self.request.user,
-            authorbalance__status=AuthorBalance.ACCEPTED,
+            status__in=[
+                Pingback.PINGBACK_TYPE_REGULAR,
+                Pingback.PINGBACK_TYPE_GOODWILL,
+                Pingback.PINGBACK_TYPE_RISK_REVIEWED_ACCEPTED,
+            ],
         ).exists()
         context['pending'] = self.object.payments.filter(
             user=self.request.user,
-            authorbalance__status=AuthorBalance.PENDING,
+            status=Pingback.PINGBACK_TYPE_RISK_UNDER_REVIEW,
         ).exists()
         return context