93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
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, ListView, TemplateView, UpdateView
|
|
from django.views.generic.edit import FormMixin
|
|
from notes.models import Note, User
|
|
|
|
|
|
# Create your views here.
|
|
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"]
|
|
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()
|
|
messages.success(self.request, "Note has been posted!")
|
|
return FormMixin.form_valid(self, form)
|
|
|
|
|
|
class ProfileView(LoginRequiredMixin, UpdateView):
|
|
model = User
|
|
fields = ["first_name", "last_name", "allow_notes_from", "expiry_seconds"]
|
|
success_url = reverse_lazy("profile")
|
|
|
|
def get_object(self):
|
|
return self.request.user
|
|
|
|
def form_valid(self, form):
|
|
ret = super().form_valid(form)
|
|
messages.success(self.request, "Profile has been updated!")
|
|
return ret
|