84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
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')
|