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