mybuddy/reports/views.py

96 lines
3.4 KiB
Python
Raw Normal View History

2017-08-21 23:21:08 +00:00
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.contrib.auth.mixins import PermissionRequiredMixin
from django.views.generic.detail import DetailView
2017-09-15 12:11:41 +00:00
from django.utils import timezone
2017-08-21 23:21:08 +00:00
2017-09-25 17:14:14 +00:00
from core.models import Child
2017-08-21 23:21:08 +00:00
2017-09-25 17:14:14 +00:00
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):
2017-10-22 19:05:58 +00:00
context = super(
DiaperChangeLifetimesChildReport, self).get_context_data(**kwargs)
2017-09-25 17:14:14 +00:00
child = context['object']
context['html'], context['javascript'] = diaperchange_lifetimes(child)
return context
2017-09-13 15:25:12 +00:00
class DiaperChangeTypesChildReport(PermissionRequiredMixin, DetailView):
"""Graph of diaper changes by day and type."""
2017-08-21 23:21:08 +00:00
model = Child
permission_required = ('core.view_child',)
2017-09-13 15:25:12 +00:00
template_name = 'reports/diaperchange_types.html'
def get_context_data(self, **kwargs):
2017-09-15 12:11:41 +00:00
context = super(DiaperChangeTypesChildReport, self).get_context_data(
**kwargs)
child = context['object']
context['html'], context['javascript'] = diaperchange_types(child)
return context
2017-09-13 15:25:12 +00:00
class SleepPatternChildReport(PermissionRequiredMixin, DetailView):
"""Graph of sleep pattern comparing sleep to wake times by day."""
model = Child
permission_required = ('core.view_child',)
2017-09-13 15:25:12 +00:00
template_name = 'reports/sleep_pattern.html'
def __init__(self):
2017-09-13 15:25:12 +00:00
super(SleepPatternChildReport, self).__init__()
self.html = ''
self.javascript = ''
def get_context_data(self, **kwargs):
2017-09-15 12:11:41 +00:00
context = super(SleepPatternChildReport, self).get_context_data(
**kwargs)
child = context['object']
2017-09-13 15:25:12 +00:00
context['html'], context['javascript'] = sleep_pattern(child)
return context
2017-09-13 15:53:32 +00:00
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):
2017-09-15 12:11:41 +00:00
context = super(SleepTotalsChildReport, self).get_context_data(
**kwargs)
2017-09-13 15:53:32 +00:00
child = context['object']
context['html'], context['javascript'] = sleep_totals(child)
return context
2017-09-15 12:11:41 +00:00
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)
2017-09-20 15:52:30 +00:00
date = self.request.GET.get('date', str(timezone.now().date()))
2017-09-16 21:17:09 +00:00
date = timezone.datetime.strptime(date, '%Y-%m-%d')
date = timezone.localtime(timezone.make_aware(date))
context['objects'] = timeline(self.object, date)
context['date'] = date
2017-09-20 15:52:30 +00:00
context['date_previous'] = date - timezone.timedelta(days=1)
context['date_next'] = date + timezone.timedelta(days=1)
2017-09-15 12:11:41 +00:00
return context