Add sent-notes and archive

server/notes/views.py

@@ -2,23 +2,63 @@ from datetime import timedelta
 from typing import Any
 from django.contrib import messages
 from django.contrib.auth.mixins import LoginRequiredMixin
+from django.core.exceptions import ImproperlyConfigured
 from django.urls import reverse_lazy
 from django.utils import timezone
-from django.views.generic import CreateView, TemplateView, UpdateView
+from django.views.generic import CreateView, ListView, TemplateView, UpdateView
 from django.views.generic.edit import FormMixin
 from notes.models import Note, User
 
 
 # Create your views here.
-class HomePage(LoginRequiredMixin, TemplateView):
+class HomePageView(LoginRequiredMixin, TemplateView):
     template_name = "notes/home.html"
 
     def get_context_data(self, **kwargs: Any) -> dict[str, Any]:
         ctx = super().get_context_data(**kwargs)
         ctx["notes"] = self.request.user.alive_received_notes
+        ctx["archive"] = self.request.user.expired_received_notes[:5]
+        ctx["alive_created"] = self.request.user.alive_created_notes
         return ctx
 
 
+class NoteView(LoginRequiredMixin, ListView):
+    # TODO: Add pagination to the template
+    template_name = "notes/note_list.html"
+    show_from = True
+    show_to = True
+
+    def get_title(self):
+        if not hasattr(self, "title"):
+            raise ImproperlyConfigured(
+                "Set a `title` attribute on subclass or override `get_title(self)`"
+            )
+        return self.title
+
+    def get_context_data(self, **kwargs: Any) -> dict[str, Any]:
+        ctx = super().get_context_data(**kwargs)
+        ctx["title"] = self.get_title()
+        ctx["show_from"] = self.show_from
+        ctx["show_to"] = self.show_to
+        return ctx
+
+
+class ArchiveView(NoteView):
+    title = "Archive"
+    show_to = False
+
+    def get_queryset(self):
+        return self.request.user.expired_received_notes
+
+
+class SentNotesView(NoteView):
+    title = "Sent Notes"
+    show_from = False
+
+    def get_queryset(self):
+        return self.request.user.all_created_notes
+
+
 class PostNoteView(LoginRequiredMixin, CreateView):
     model = Note
     fields = ["to_user", "note"]