notes: updated the form to have two buttons and not require a post

1 files changed, 9 insertions, 2 deletions

diff --git a/app/notes/views.py b/app/notes/views.py
index 53312ef..fd6aa68 100644
--- a/app/notes/views.py
+++ b/app/notes/views.py
@@ -16,8 +16,15 @@ class NoteCreateView(LoginRequiredMixin, CreateView):
        kwargs.update({'user': self.request.user})
        return kwargs
 
-    def get_success_url(self):
-        return reverse('posts:detail', kwargs={"pk": self.object.post.pk})
+    def get_Success_url(self):
+        if 'add_new' in self.request.POST:
+            return reverse('notes:create')
+        else:
+            if self.object.post:
+                return reverse('posts:detail', kwargs={"pk": self.object.post.pk})
+            else:
+                return reverse('notes:create')
+
 
 
 class NoteUpdateView(LoginRequiredMixin, UpdateView):