82 lines
2.8 KiB
Python
Executable File
82 lines
2.8 KiB
Python
Executable File
from django.db.models import Q, F, Sum, Count, ExpressionWrapper, IntegerField
|
|
from django.db.models.functions import Substr
|
|
from django.views.generic import TemplateView
|
|
|
|
from app.base.models import Booking, BookingType, Trait, TraitMapping
|
|
from app.base.views.login import LoginRequired
|
|
from app.base.views.model_views.base import ViewOptions
|
|
|
|
from datetime import date
|
|
from typing import Any
|
|
|
|
|
|
class DashboardOptions(ViewOptions, LoginRequired):
|
|
icon = 'chart-line'
|
|
title = 'Dashboard'
|
|
|
|
|
|
class DashboardView(DashboardOptions, TemplateView):
|
|
template_name = 'dashboard.html'
|
|
|
|
def get_context_data(self, **kwargs) -> 'dict[str, Any]':
|
|
context = super().get_context_data(**kwargs)
|
|
book_types = dict(BookingType.objects.values_list('key', 'label'))
|
|
trait_types = dict(Trait.objects.values_list('key', 'label'))
|
|
|
|
# Booking stats
|
|
|
|
book_by_type = stats_for_booking('type')
|
|
book_by_month = {}
|
|
|
|
for stat in stats_for_booking('month'):
|
|
year, month = stat['month'].split('-')
|
|
if year not in book_by_month:
|
|
book_by_month[year] = [[0, 0] for x in range(13)]
|
|
book_by_month[year][0][0] += stat['sum']
|
|
book_by_month[year][0][1] += stat['count']
|
|
book_by_month[year][int(month)][0] += stat['sum']
|
|
book_by_month[year][int(month)][1] += stat['count']
|
|
|
|
context['booking'] = {
|
|
'labels': book_types,
|
|
'by_type': book_by_type,
|
|
'by_month': book_by_month,
|
|
}
|
|
context['trait'] = {
|
|
'labels': trait_types,
|
|
'by_type': stats_for_traits(),
|
|
}
|
|
context['head'] = {
|
|
'bookings': Booking._meta.verbose_name_plural,
|
|
'booking_types': BookingType._meta.verbose_name_plural,
|
|
'traits': Trait._meta.verbose_name_plural,
|
|
}
|
|
|
|
return context
|
|
|
|
|
|
def stats_for_booking(groupby: str):
|
|
_Q = Booking.objects.filter(Q(end_time__isnull=False)).annotate(
|
|
# for whatever reason django uses nano seconds
|
|
diff=ExpressionWrapper(
|
|
(F('end_time') - F('begin_time')) / 60_000_000,
|
|
output_field=IntegerField()),
|
|
).filter(diff__lt=1440) # filter 24+ hours. Probably wrong checkout
|
|
|
|
if groupby == 'month':
|
|
_Q = _Q.annotate(month=Substr('begin_time', 1, 7)) # YYYY-MM
|
|
|
|
return _Q.values(groupby).annotate(
|
|
sum=Sum('diff', output_field=IntegerField()),
|
|
count=Count(1),
|
|
).values(groupby, 'sum', 'count').order_by(groupby)
|
|
|
|
|
|
def stats_for_traits():
|
|
return TraitMapping.objects.filter(
|
|
Q(valid_from__lte=date.today()),
|
|
Q(valid_until=None) | Q(valid_until__gte=date.today()),
|
|
).values('trait').annotate(
|
|
count=Count(1),
|
|
).values('trait', 'count').order_by('trait')
|