1 files changed, 4 insertions, 16 deletions
diff --git a/app/notes/views.py b/app/notes/views.py
index 7c8f6d8..53312ef 100644
--- a/app/notes/views.py
+++ b/app/notes/views.py
@@ -1,14 +1,13 @@
 from django.views.generic import UpdateView, DetailView, ListView 
 from django.views.generic.edit import CreateView, DeleteView, UpdateView
-from django.utils.decorators import method_decorator
-from django.contrib.auth.decorators import login_required
+from django.contrib.auth.mixins import LoginRequiredMixin
 from django.urls import reverse
 
 from .models import Note
 from .forms import NoteCreateForm
 
 
-class NoteCreateView(CreateView):
+class NoteCreateView(LoginRequiredMixin, CreateView):
     model = Note
     form_class = NoteCreateForm
 
@@ -20,11 +19,8 @@ class NoteCreateView(CreateView):
     def get_success_url(self):
         return reverse('posts:detail', kwargs={"pk": self.object.post.pk})
 
-    @method_decorator(login_required)
-    def dispatch(self, *args, **kwargs):
-        return super(NoteCreateView, self).dispatch(*args, **kwargs)
 
-class NoteUpdateView(UpdateView):
+class NoteUpdateView(LoginRequiredMixin, UpdateView):
     model = Note
     fields = ['title', 'url', 'body_markdown', 'post', 'status', 'user']
     
@@ -36,11 +32,8 @@ class NoteUpdateView(UpdateView):
     def get_success_url(self):
         return reverse('posts:detail', kwargs={"pk": self.object.post.pk})
 
-    @method_decorator(login_required)
-    def dispatch(self, *args, **kwargs):
-        return super(NoteUpdateView, self).dispatch(*args, **kwargs)
 
-class NoteListView(ListView):
+class NoteListView(LoginRequiredMixin, ListView):
     model = Note
 
     def get_queryset(self):
@@ -53,8 +46,3 @@ class NoteListView(ListView):
             qs = Note.objects.filter(user=self.request.user).filter(status=status_reverse[status])
         qs = Note.objects.filter(user=self.request.user)
         return qs
-
-
-    @method_decorator(login_required)
-    def dispatch(self, *args, **kwargs):
-        return super(NoteListView, self).dispatch(*args, **kwargs)