From be4b5904a575390787b349f05f3c8dd5ad4bddb2 Mon Sep 17 00:00:00 2001 From: luxagraf Date: Mon, 24 Jul 2023 16:23:42 -0500 Subject: posts/notes: cleaned up view code and restyled table view --- app/notes/views.py | 20 ++++---------------- 1 file changed, 4 insertions(+), 16 deletions(-) (limited to 'app/notes') 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) -- cgit v1.2.3-70-g09d2