Show 404 if course not found in CourseBuySuccessView

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})