58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
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
|
|
|
|
|
|
# Create your views here.
|
|
|
|
|
|
class ListPosts(ListView):
|
|
model = Post
|
|
template_name = "list_posts.html"
|
|
paginate_by = 3
|
|
|
|
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 get_queryset(self) -> QuerySet[Any]:
|
|
# queryset = Post.objects.get(pk=self.request.GET.get("pk"))
|
|
|
|
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)
|