khaganat-web/logs/views.py

from django.db.models.functions import TruncDate
from django.db.models import Count
from django.views import generic
from django.conf import settings
from .models import Source, Entry
import datetime


class IndexView(generic.ListView):
    template_name = 'logs/index.html'
    context_object_name = 'sources'

    def get_queryset(self):
        now = datetime.date.today()
        start_date = now - datetime.timedelta(
            days=settings.KHAGANAT_LOGS_MAX_DAYS
        )
        end_date = now - datetime.timedelta(
            days=settings.KHAGANAT_LOGS_MIN_DAYS - 1
        )
        out = []
        qs = Entry.objects.filter(
            hidden=False,
            source__hidden=False,
            created__range=(start_date, end_date)
        ).values('source', 'source__name', 'source__slug').annotate(
            nb=Count('id')
        )

        for src in qs:
            src['stats'] = Entry.objects.filter(
                hidden=False,
                source=src['source'],
                created__range=(start_date, end_date)
            ).annotate(
                date=TruncDate('created')
            ).values('date').annotate(nb=Count('date')).order_by('-date')
            out.append(src)

        return out


class EntriesView(generic.ListView):
    context_object_name = 'entries'
    template_name = 'logs/entries.html'
    allow_empty = False

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['date'] = datetime.date(
            self.kwargs['year'],
            self.kwargs['month'],
            self.kwargs['day']
        )
        context['source_channel'] = Source.objects.get(
            slug=self.kwargs['source']
        )
        if isinstance(context['source_channel'], tuple):
            context['source_channel'] = context['source_channel'][0]
        return context

    def get_queryset(self):
        dt = datetime.date(
            self.kwargs['year'],
            self.kwargs['month'],
            self.kwargs['day']
        )
        now = datetime.date.today()
        out_of_bounds = any((
            (now - dt).days > settings.KHAGANAT_LOGS_MAX_DAYS,
            (now - dt).days < settings.KHAGANAT_LOGS_MIN_DAYS,
        ))
        if out_of_bounds:
            return Entry.objects.none()
        src = Source.objects.get(slug=self.kwargs['source'])
        if src is None or src.hidden:
            return Entry.objects.none()
        return Entry.objects.filter(
            hidden=False,
            source=src,
            created__year=dt.year,
            created__month=dt.month,
            created__day=dt.day
        ).order_by('created')