views.py

from django.views.generic.dates import (
    ArchiveIndexView,
    DateDetailView,
    DayArchiveView,
    MonthArchiveView,
    YearArchiveView,
)

from .models import Entry, Event


class BlogViewMixin:
    date_field = "pub_date"
    paginate_by = 10

    def get_allow_future(self):
        return self.request.user.is_staff

    def get_queryset(self):
        return Entry.objects.published()

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)

        events_queryset = Event.objects.future().published()

        context["events"] = events_queryset[:3]

        return context


class BlogArchiveIndexView(BlogViewMixin, ArchiveIndexView):
    pass


class BlogYearArchiveView(BlogViewMixin, YearArchiveView):
    make_object_list = True


class BlogMonthArchiveView(BlogViewMixin, MonthArchiveView):
    pass


class BlogDayArchiveView(BlogViewMixin, DayArchiveView):
    pass


class BlogDateDetailView(BlogViewMixin, DateDetailView):

    def get_queryset(self):
        """Allows staff users to view unpublished entries"""
        if self.request.user.is_staff:
            print("\n\nSTAFF USER\n\n")
            return Entry.objects.all()
        else:
            print("\n\nNORMAL USER\n\n")
            return Entry.objects.published()