diff --git a/purchase/views.py b/purchase/views.py index 91ea44a..51402c6 100644 --- a/purchase/views.py +++ b/purchase/views.py @@ -1,3 +1,21 @@ -from django.shortcuts import render +from django.shortcuts import redirect, render +from django.urls import reverse -# Create your views here. +from home.models import CoursePage +from purchase.models import CoursePurchase + + +def mock_purchase_course(request, course_id): + course = CoursePage.objects.get(id=course_id) + + course.mock_purchase(request.user) + + return redirect(course.url) + + +def mock_refund_purchase(request, purchase_id): + purchase = CoursePurchase.objects.get(id=purchase_id) + + purchase.mock_refund() + + return redirect(purchase.course.url)