mybuddy/reports/views.py

# -*- coding: utf-8 -*-
from django.views.generic.detail import DetailView

from babybuddy.mixins import PermissionRequiredMixin
from core import models

from . import graphs


class DiaperChangeAmounts(PermissionRequiredMixin, DetailView):
    """
    Graph of diaper "amounts" - measurements of urine output.
    """
    model = models.Child
    permission_required = ('core.view_child',)
    template_name = 'reports/diaperchange_amounts.html'

    def get_context_data(self, **kwargs):
        context = super(DiaperChangeAmounts, self).get_context_data(**kwargs)
        child = context['object']
        changes = models.DiaperChange.objects.filter(child=child, amount__gt=0)
        if changes and changes.count() > 0:
            context['html'], context['js'] = \
                graphs.diaperchange_amounts(changes)
        return context


class DiaperChangeLifetimesChildReport(PermissionRequiredMixin, DetailView):
    """
    Graph of diaper "lifetimes" - time between diaper changes.
    """
    model = models.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']
        changes = models.DiaperChange.objects.filter(child=child)
        if changes and changes.count() > 1:
            context['html'], context['js'] = \
                graphs.diaperchange_lifetimes(changes)
        return context


class DiaperChangeTypesChildReport(PermissionRequiredMixin, DetailView):
    """
    Graph of diaper changes by day and type.
    """
    model = models.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']
        changes = models.DiaperChange.objects.filter(child=child)
        if changes:
            context['html'], context['js'] = \
                graphs.diaperchange_types(changes)
        return context


class FeedingAmountsChildReport(PermissionRequiredMixin, DetailView):
    """
    Graph of daily feeding amounts over time.
    """
    model = models.Child
    permission_required = ('core.view_child',)
    template_name = 'reports/feeding_amounts.html'

    def __init__(self):
        super(FeedingAmountsChildReport, self).__init__()
        self.html = ''
        self.js = ''

    def get_context_data(self, **kwargs):
        context = super(FeedingAmountsChildReport, self).get_context_data(
            **kwargs)
        child = context['object']
        instances = models.Feeding.objects.filter(child=child)
        if instances:
            context['html'], context['js'] = graphs.feeding_amounts(instances)
        return context


class FeedingDurationChildReport(PermissionRequiredMixin, DetailView):
    """
    Graph of feeding durations over time.
    """
    model = models.Child
    permission_required = ('core.view_child',)
    template_name = 'reports/feeding_duration.html'

    def __init__(self):
        super(FeedingDurationChildReport, self).__init__()
        self.html = ''
        self.js = ''

    def get_context_data(self, **kwargs):
        context = super(FeedingDurationChildReport, self).get_context_data(
            **kwargs)
        child = context['object']
        instances = models.Feeding.objects.filter(child=child)
        if instances:
            context['html'], context['js'] = graphs.feeding_duration(instances)
        return context


class SleepPatternChildReport(PermissionRequiredMixin, DetailView):
    """
    Graph of sleep pattern comparing sleep to wake times by day.
    """
    model = models.Child
    permission_required = ('core.view_child',)
    template_name = 'reports/sleep_pattern.html'

    def __init__(self):
        super(SleepPatternChildReport, self).__init__()
        self.html = ''
        self.js = ''

    def get_context_data(self, **kwargs):
        context = super(SleepPatternChildReport, self).get_context_data(
            **kwargs)
        child = context['object']
        instances = models.Sleep.objects.filter(child=child).order_by('start')
        if instances:
            context['html'], context['js'] = graphs.sleep_pattern(instances)
        return context


class SleepTotalsChildReport(PermissionRequiredMixin, DetailView):
    """
    Graph of total sleep by day.
    """
    model = models.Child
    permission_required = ('core.view_child',)
    template_name = 'reports/sleep_totals.html'

    def __init__(self):
        super(SleepTotalsChildReport, self).__init__()
        self.html = ''
        self.js = ''

    def get_context_data(self, **kwargs):
        context = super(SleepTotalsChildReport, self).get_context_data(
            **kwargs)
        child = context['object']
        instances = models.Sleep.objects.filter(child=child).order_by('start')
        if instances:
            context['html'], context['js'] = graphs.sleep_totals(instances)
        return context


class WeightWeightChildReport(PermissionRequiredMixin, DetailView):
    """
    Graph of weight change over time.
    """
    model = models.Child
    permission_required = ('core.view_child',)
    template_name = 'reports/weight_change.html'

    def get_context_data(self, **kwargs):
        context = super(WeightWeightChildReport, self).get_context_data(
            **kwargs)
        child = context['object']
        objects = models.Weight.objects.filter(child=child)
        if objects:
            context['html'], context['js'] = graphs.weight_weight(objects)
        return context