34 lines
913 B
Python
34 lines
913 B
Python
from django.views.generic import ListView, DetailView
|
|
from django.utils import timezone
|
|
from . import models
|
|
|
|
|
|
class NewsIndex(ListView):
|
|
model = models.NewsItem
|
|
template_name = 'news_index.html'
|
|
context_object_name = 'news_items'
|
|
|
|
def get_queryset(self):
|
|
return self.model.objects.public()
|
|
|
|
|
|
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()
|
|
timed = news_item.published_at > timezone.now()
|
|
|
|
if news_item.public and timed:
|
|
context['not_public'] = True
|
|
context['timed'] = True
|
|
elif not news_item.public:
|
|
context['not_public'] = True
|
|
context['timed'] = False
|
|
|
|
return context
|
|
|