~widelands-dev/widelands-website/django_staticfiles

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
from django.shortcuts import get_object_or_404
from news.models import Post, Category
from django.views.generic import \
    ListView, \
    ArchiveIndexView, \
    YearArchiveView, \
    MonthArchiveView, \
    DateDetailView


class NewsList(ArchiveIndexView):

    template_name = 'news/category_posts.html'
    model = Post
    date_field = 'publish'

    def get_context_data(self, **kwargs):
        # Call the base implementation first to get a context
        context = super(NewsList, self).get_context_data(**kwargs)

        context['categories'] = Category.objects.all()
        return context


class YearNews(YearArchiveView):

    model = Post
    template_name = 'news/post_archive_year.html'
    date_field = 'publish'
    make_object_list = True


class MonthNews(MonthArchiveView):

    model = Post
    template_name = 'news/post_archive_month.html'
    date_field = 'publish'


class NewsDetail(DateDetailView):

    model = Post
    template_name = 'news/post_detail.html'
    date_field = 'publish'


class CategoryView(ListView):

    template_name = 'news/category_posts.html'

    def get_context_data(self, **kwargs):
        # Call the base implementation first to get a context
        context = super(CategoryView, self).get_context_data(**kwargs)
        if self.kwargs['slug'] == 'none':
            # Exemption for posts with no category
            context['cur_category'] = 'None'
        else:
            context['cur_category'] = get_object_or_404(
                Category, slug=self.kwargs['slug'])

        context['categories'] = Category.objects.all()
        return context

    def get_queryset(self):
        # Gather posts filtered by category
        if self.kwargs['slug'] == 'none':
            # Posts mustn't have a category
            qs = Post.objects.published().filter(categories=None)
        else:
            qs = Post.objects.published().filter(
                categories__slug=self.kwargs['slug'])
        return qs