66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
from typing import Union
|
|
|
|
from django.conf import settings
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.db.models import F
|
|
from django.views import generic
|
|
|
|
from articles.models import Article, Page
|
|
|
|
|
|
class BaseArticleListView(generic.ListView):
|
|
paginate_by = 10
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context["blog_title"] = settings.BLOG["title"]
|
|
context["blog_description"] = settings.BLOG["description"]
|
|
return context
|
|
|
|
|
|
class ArticlesListView(BaseArticleListView):
|
|
model = Article
|
|
context_object_name = "articles"
|
|
queryset = Article.without_pages.filter(status=Article.PUBLISHED)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
index_page = Page.objects.filter(
|
|
status=Article.PUBLISHED, position=0
|
|
).first() # type: Page
|
|
context["article"] = index_page
|
|
return context
|
|
|
|
|
|
class DraftsListView(LoginRequiredMixin, BaseArticleListView):
|
|
model = Article
|
|
context_object_name = "articles"
|
|
queryset = Article.objects.filter(status=Article.DRAFT)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context["title"] = "Drafts"
|
|
context["title_header"] = context["title"]
|
|
return context
|
|
|
|
|
|
class ArticleDetailView(generic.DetailView):
|
|
model = Article
|
|
context_object_name = "article"
|
|
template_name = "articles/article_detail.html"
|
|
|
|
def get_queryset(self):
|
|
queryset = super().get_queryset()
|
|
if self.request.user.is_authenticated:
|
|
return queryset
|
|
return queryset.filter(status=Article.PUBLISHED)
|
|
|
|
def get_object(self, queryset=None) -> Union[Article, Page]:
|
|
obj = super().get_object(queryset) # type: Article
|
|
if hasattr(obj, "page"):
|
|
obj = obj.page # type: Page
|
|
if not self.request.user.is_authenticated:
|
|
obj.views_count = F("views_count") + 1
|
|
obj.save(update_fields=["views_count"])
|
|
|
|
return obj
|