Plankton_Detector/PlanktonDetector/Community/views.py

from typing import Any
from django.db.models.query import QuerySet
from django.http import HttpResponse
from django.shortcuts import redirect, render
from django.views.generic import ListView, DetailView
from django.views.generic.edit import FormView, FormMixin
from .models import Post, Comment
from .froms import PostForm, CommentForm


class ListPosts(ListView):
    model = Post
    template_name = "list_posts.html"
    paginate_by = 4

    def get_queryset(self) -> QuerySet[Any]:
        queryset = Post.objects.all().order_by("-date_pub")
        return queryset


class PostDetails(DetailView, FormMixin):
    model = Post
    template_name = "post_details.html"
    form_class = CommentForm

    def post(self, request, *args, **kwargs):
        self.object = self.get_object()
        form = self.get_form()
        if form.is_valid():
            return self.form_valid(form)
        else:
            return self.form_invalid(form)

    def form_valid(self, form) -> HttpResponse:
        post = self.get_object()
        new_comment = form.save(commit=False)
        new_comment.author = self.request.user
        new_comment.post = post
        new_comment.save()
        return redirect("post-details", pk=post.id)


class AddPost(FormView):
    template_name = "add_post.html"
    form_class = PostForm

    def form_valid(self, form) -> HttpResponse:
        new_post = form.save(commit=False)
        new_post.author = self.request.user
        new_post.save()
        return redirect("post-details", pk=new_post.id)