1 files changed, 14 insertions, 1 deletions
diff --git a/app/posts/views.py b/app/posts/views.py
index e1af9b5..6a5ff95 100644
--- a/app/posts/views.py
+++ b/app/posts/views.py
@@ -38,7 +38,7 @@ class NoteCreateView(CreateView):
 
 class NoteUpdateView(UpdateView):
     model = Note
-    fields = ['title', 'url', 'body_markdown', 'post', 'status']
+    fields = ['title', 'url', 'body_markdown', 'post', 'status', 'user']
     
     def get_context_data(self, **kwargs):
         context = super(NoteUpdateView, self).get_context_data(**kwargs)
@@ -47,6 +47,19 @@ class NoteUpdateView(UpdateView):
 
     def get_success_url(self):
         return reverse('posts:detail', kwargs={"pk": self.object.post.pk})
+
+
+class NoteListView(ListView):
+    model = Note
+
+    def get_queryset(self):
+        if self.kwargs['status'] == "todo":
+            status = "Call In"
+        else:
+            status = self.kwargs['status']
+        status_reverse = dict((v, k) for k, v in Note.STATUS)
+        return Note.objects.filter(user=self.request.user).filter(status=status_reverse[status])
+
 '''
 
 class UpdateViewWithUser(UpdateView):