43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from django.shortcuts import get_object_or_404
|
|
from django.views.generic import DetailView, ListView
|
|
|
|
from .models import Article, Category
|
|
|
|
|
|
class ArticleList(ListView):
|
|
category = None
|
|
tag = None
|
|
paginate_by = 9
|
|
|
|
def get_queryset(self):
|
|
category_slug = self.kwargs.get('category', None)
|
|
self.tag = self.kwargs.get('tag', None)
|
|
if category_slug:
|
|
self.category = get_object_or_404(Category, slug=category_slug)
|
|
return Article.objects.filter(category=self.category).filter(is_published=True)
|
|
elif self.tag:
|
|
return Article.objects.filter(tags=self.tag).filter(is_published=True)
|
|
else:
|
|
return Article.objects.filter(is_published=True)
|
|
|
|
def get_context_data(self, *, object_list=None, **kwargs):
|
|
context = super(ArticleList, self).get_context_data(**kwargs)
|
|
context['category'] = self.category
|
|
context['tag'] = self.tag
|
|
return context
|
|
|
|
|
|
class ArticleDetail(DetailView):
|
|
model = Article
|
|
|
|
|
|
class CategoryList(ListView):
|
|
model = Category
|
|
|
|
|
|
class TagList(ListView):
|
|
template_name = 'articles/tag_list.html'
|
|
|
|
def get_queryset(self):
|
|
return Article.tags.tag_model.objects.all()
|