mybuddy/reports/views.py

96 lines
3.4 KiB
Python

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.contrib.auth.mixins import PermissionRequiredMixin
from django.views.generic.detail import DetailView
from django.utils import timezone
from core.models import Child
from .graphs import (diaperchange_types, diaperchange_lifetimes, sleep_pattern,
sleep_totals, timeline)
class DiaperChangeLifetimesChildReport(PermissionRequiredMixin, DetailView):
"""Graph of diaper changes by day and type."""
model = Child
permission_required = ('core.view_child',)
template_name = 'reports/diaperchange_lifetimes.html'
def get_context_data(self, **kwargs):
context = super(DiaperChangeLifetimesChildReport, self).get_context_data(
**kwargs)
child = context['object']
context['html'], context['javascript'] = diaperchange_lifetimes(child)
return context
class DiaperChangeTypesChildReport(PermissionRequiredMixin, DetailView):
"""Graph of diaper changes by day and type."""
model = Child
permission_required = ('core.view_child',)
template_name = 'reports/diaperchange_types.html'
def get_context_data(self, **kwargs):
context = super(DiaperChangeTypesChildReport, self).get_context_data(
**kwargs)
child = context['object']
context['html'], context['javascript'] = diaperchange_types(child)
return context
class SleepPatternChildReport(PermissionRequiredMixin, DetailView):
"""Graph of sleep pattern comparing sleep to wake times by day."""
model = Child
permission_required = ('core.view_child',)
template_name = 'reports/sleep_pattern.html'
def __init__(self):
super(SleepPatternChildReport, self).__init__()
self.html = ''
self.javascript = ''
def get_context_data(self, **kwargs):
context = super(SleepPatternChildReport, self).get_context_data(
**kwargs)
child = context['object']
context['html'], context['javascript'] = sleep_pattern(child)
return context
class SleepTotalsChildReport(PermissionRequiredMixin, DetailView):
"""Graph of total sleep by day."""
model = Child
permission_required = ('core.view_child',)
template_name = 'reports/sleep_totals.html'
def __init__(self):
super(SleepTotalsChildReport, self).__init__()
self.html = ''
self.javascript = ''
def get_context_data(self, **kwargs):
context = super(SleepTotalsChildReport, self).get_context_data(
**kwargs)
child = context['object']
context['html'], context['javascript'] = sleep_totals(child)
return context
class TimelineChildReport(PermissionRequiredMixin, DetailView):
"""Graph of total sleep by day."""
model = Child
permission_required = ('core.view_child',)
template_name = 'reports/timeline.html'
def get_context_data(self, **kwargs):
context = super(TimelineChildReport, self).get_context_data(**kwargs)
date = self.request.GET.get('date', str(timezone.now().date()))
date = timezone.datetime.strptime(date, '%Y-%m-%d')
date = timezone.localtime(timezone.make_aware(date))
context['objects'] = timeline(self.object, date)
context['date'] = date
context['date_previous'] = date - timezone.timedelta(days=1)
context['date_next'] = date + timezone.timedelta(days=1)
return context