26 lines
744 B
Python
26 lines
744 B
Python
from django.views.generic import ListView, DetailView
|
|
from django.utils import timezone
|
|
from . import models
|
|
|
|
|
|
class NewsIndex(ListView):
|
|
model = models.NewsItem
|
|
queryset = models.NewsItem.objects.public()
|
|
template_name = 'news_index.html'
|
|
context_object_name = 'news_items'
|
|
|
|
|
|
class NewsDetail(DetailView):
|
|
model = models.NewsItem
|
|
template_name = 'news_detail.html'
|
|
context_object_name = 'news_item'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(NewsDetail, self).get_context_data(**kwargs)
|
|
news_item = self.get_object()
|
|
context['draft'] = False
|
|
if news_item.public and news_item.published_at > timezone.now():
|
|
context['draft'] = True
|
|
return context
|
|
|