Add Post a Note page

notes/views.py

@@ -1,6 +1,11 @@
+from datetime import timedelta
 from typing import Any
 from django.contrib.auth.mixins import LoginRequiredMixin
-from django.views.generic import TemplateView
+from django.urls import reverse_lazy
+from django.utils import timezone
+from django.views.generic import CreateView, TemplateView
+from django.views.generic.edit import FormMixin
+from notes.models import Note
 
 # Create your views here.
 class HomePage(LoginRequiredMixin, TemplateView):
@@ -10,3 +15,21 @@ class HomePage(LoginRequiredMixin, TemplateView):
         ctx = super().get_context_data(**kwargs)
         ctx['notes'] = self.request.user.alive_received_notes
         return ctx
+
+
+class PostNoteView(LoginRequiredMixin, CreateView):
+    model = Note
+    fields = ["to_user", "note"]
+    success_url = reverse_lazy("post-a-note")
+
+    def get_form(self, form_class=None):
+        form = super().get_form(form_class)
+        form.fields['to_user'].queryset = self.request.user.allowed_notes_to.all()
+        return form
+
+    def form_valid(self, form):
+        self.object = note = form.save(commit=False)
+        note.expiry = timezone.now() + timedelta(seconds=note.to_user.expiry_seconds)
+        note.from_user = self.request.user
+        note.save()
+        return FormMixin.form_valid(self, form)