47 - Custom Mixin for Class Based Views

Author: codingforentrepreneurs

src/courses/views.py

@@ -5,16 +5,17 @@
 from .models import Course 
 # BASE VIEW CLass = VIEW
 
-
-class CourseDeleteView(View):
-    template_name = "courses/course_delete.html" # DetailView
+class CourseObjectMixin(object):
+    model = Course
     def get_object(self):
         id = self.kwargs.get('id')
         obj = None
         if id is not None:
-            obj = get_object_or_404(Course, id=id)
-        return obj
+            obj = get_object_or_404(self.model, id=id)
+        return obj 
 
+class CourseDeleteView(CourseObjectMixin, View):
+    template_name = "courses/course_delete.html" # DetailView
     def get(self, request, id=None, *args, **kwargs):
         # GET method
         context = {}
@@ -34,7 +35,7 @@ def post(self, request, id=None,  *args, **kwargs):
         return render(request, self.template_name, context)
 
 
-class CourseUpdateView(View):
+class CourseUpdateView(CourseObjectMixin, View):
     template_name = "courses/course_update.html" # DetailView
     def get_object(self):
         id = self.kwargs.get('id')
@@ -96,14 +97,11 @@ def get(self, request, *args, **kwargs):
         return render(request, self.template_name, context)
 
 
-class CourseView(View):
+class CourseView(CourseObjectMixin, View):
     template_name = "courses/course_detail.html" # DetailView
     def get(self, request, id=None, *args, **kwargs):
         # GET method
-        context = {}
-        if id is not None:
-            obj = get_object_or_404(Course, id=id)
-            context['object'] = obj
+        context = {'object': self.get_object()}
         return render(request, self.template_name, context)
 
     # def post(request, *args, **kwargs):