diff --git a/apps/payment/views.py b/apps/payment/views.py index 10392769..3e3165ab 100644 --- a/apps/payment/views.py +++ b/apps/payment/views.py @@ -6,7 +6,7 @@ from datetime import timedelta from django.contrib import messages from django.contrib.auth.decorators import login_required from django.http import HttpResponse -from django.shortcuts import redirect +from django.shortcuts import redirect, get_object_or_404 from django.views.generic import View, TemplateView from django.views.decorators.csrf import csrf_exempt from django.urls import reverse_lazy @@ -28,7 +28,7 @@ class CourseBuySuccessView(TemplateView): template_name = 'payment/course_payment_success.html' def get(self, request, pk=None, *args, **kwargs): - course = Course.objects.get(id=pk) + course = get_object_or_404(Course, pk=pk) return self.render_to_response(context={'course': course})