posts: added todo view of guides and reviews that need to be done

2 files changed, 8 insertions, 1 deletions

diff --git a/app/notes/models.py b/app/notes/models.py
index ffd0a25..7f8dab0 100644
--- a/app/notes/models.py
+++ b/app/notes/models.py
@@ -22,6 +22,13 @@ class Note(models.Model):
         (4, 'Live'),
     )
     status = models.IntegerField(choices=STATUS, default=0)
+    PLAN = (
+        (0, 'For Guide'),
+        (1, 'Review'),
+        (2, 'Rave'),
+        (3, 'Rant'),
+    )
+    plan = models.IntegerField(choices=PLAN, default=0)
 
     class Meta:
         ordering = ('date_created', 'status')
diff --git a/app/notes/views.py b/app/notes/views.py
index 7dd9c04..209d896 100644
--- a/app/notes/views.py
+++ b/app/notes/views.py
@@ -29,7 +29,7 @@ class NoteCreateView(LoginRequiredMixin, CreateView):
 
 class NoteUpdateView(LoginRequiredMixin, UpdateView):
     model = Note
-    fields = ['title', 'url', 'body_markdown', 'post', 'status', 'user']
+    fields = ['title', 'url', 'body_markdown', 'post', 'status', 'plan']
     
     def get_context_data(self, **kwargs):
         context = super(NoteUpdateView, self).get_context_data(**kwargs)