mirror of https://github.com/snachodog/mybuddy.git
540 lines
18 KiB
Python
540 lines
18 KiB
Python
# -*- coding: utf-8 -*-
|
|
from django.contrib import messages
|
|
from django.contrib.messages.views import SuccessMessageMixin
|
|
from django.forms import Form
|
|
from django.http import HttpResponseRedirect
|
|
from django.urls import reverse, reverse_lazy
|
|
from django.utils import timezone
|
|
from django.utils.translation import gettext as _
|
|
from django.views.generic.base import RedirectView, TemplateView
|
|
from django.views.generic.detail import DetailView
|
|
from django.views.generic.edit import CreateView, UpdateView, DeleteView, FormView
|
|
|
|
from babybuddy.mixins import LoginRequiredMixin, PermissionRequiredMixin
|
|
from babybuddy.views import BabyBuddyFilterView
|
|
from core import filters, forms, models, timeline
|
|
|
|
|
|
def _prepare_timeline_context_data(context, date, child=None):
|
|
date = timezone.datetime.strptime(date, "%Y-%m-%d")
|
|
date = timezone.localtime(timezone.make_aware(date))
|
|
context["timeline_objects"] = timeline.get_objects(date, child)
|
|
context["date"] = date
|
|
context["date_previous"] = date - timezone.timedelta(days=1)
|
|
if date.date() < timezone.localdate():
|
|
context["date_next"] = date + timezone.timedelta(days=1)
|
|
pass
|
|
|
|
|
|
class CoreAddView(PermissionRequiredMixin, SuccessMessageMixin, CreateView):
|
|
def get_success_message(self, cleaned_data):
|
|
cleaned_data["model"] = self.model._meta.verbose_name.title()
|
|
if "child" in cleaned_data:
|
|
self.success_message = _("%(model)s entry for %(child)s added!")
|
|
else:
|
|
self.success_message = _("%(model)s entry added!")
|
|
return self.success_message % cleaned_data
|
|
|
|
def get_form_kwargs(self):
|
|
"""
|
|
Check for and add "child" and "timer" from request query parameters.
|
|
- "child" may provide a slug for a Child instance.
|
|
- "timer" may provided an ID for a Timer instance.
|
|
|
|
These arguments are used in some add views to pre-fill initial data in
|
|
the form fields.
|
|
|
|
:return: Updated keyword arguments.
|
|
"""
|
|
kwargs = super(CoreAddView, self).get_form_kwargs()
|
|
for parameter in ["child", "timer"]:
|
|
value = self.request.GET.get(parameter, None)
|
|
if value:
|
|
kwargs.update({parameter: value})
|
|
return kwargs
|
|
|
|
|
|
class CoreUpdateView(PermissionRequiredMixin, SuccessMessageMixin, UpdateView):
|
|
def get_success_message(self, cleaned_data):
|
|
cleaned_data["model"] = self.model._meta.verbose_name.title()
|
|
if "child" in cleaned_data:
|
|
self.success_message = _("%(model)s entry for %(child)s updated.")
|
|
else:
|
|
self.success_message = _("%(model)s entry updated.")
|
|
return self.success_message % cleaned_data
|
|
|
|
|
|
class CoreDeleteView(PermissionRequiredMixin, SuccessMessageMixin, DeleteView):
|
|
def get_success_message(self, cleaned_data):
|
|
return _("%(model)s entry deleted.") % {
|
|
"model": self.model._meta.verbose_name.title()
|
|
}
|
|
|
|
|
|
class BMIList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.BMI
|
|
template_name = "core/bmi_list.html"
|
|
permission_required = ("core.view_bmi",)
|
|
paginate_by = 10
|
|
filterset_class = filters.BMIFilter
|
|
|
|
|
|
class BMIAdd(CoreAddView):
|
|
model = models.BMI
|
|
permission_required = ("core.add_bmi",)
|
|
form_class = forms.BMIForm
|
|
success_url = reverse_lazy("core:bmi-list")
|
|
|
|
|
|
class BMIUpdate(CoreUpdateView):
|
|
model = models.BMI
|
|
permission_required = ("core.change_bmi",)
|
|
form_class = forms.BMIForm
|
|
success_url = reverse_lazy("core:bmi-list")
|
|
|
|
|
|
class BMIDelete(CoreDeleteView):
|
|
model = models.BMI
|
|
permission_required = ("core.delete_bmi",)
|
|
success_url = reverse_lazy("core:bmi-list")
|
|
|
|
|
|
class ChildList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Child
|
|
template_name = "core/child_list.html"
|
|
permission_required = ("core.view_child",)
|
|
paginate_by = 10
|
|
filterset_fields = ("first_name", "last_name")
|
|
|
|
|
|
class ChildAdd(CoreAddView):
|
|
model = models.Child
|
|
permission_required = ("core.add_child",)
|
|
form_class = forms.ChildForm
|
|
success_url = reverse_lazy("core:child-list")
|
|
success_message = _("%(first_name)s %(last_name)s added!")
|
|
|
|
|
|
class ChildDetail(PermissionRequiredMixin, DetailView):
|
|
model = models.Child
|
|
permission_required = ("core.view_child",)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(ChildDetail, self).get_context_data(**kwargs)
|
|
date = self.request.GET.get("date", str(timezone.localdate()))
|
|
_prepare_timeline_context_data(context, date, self.object)
|
|
return context
|
|
|
|
|
|
class ChildUpdate(CoreUpdateView):
|
|
model = models.Child
|
|
permission_required = ("core.change_child",)
|
|
form_class = forms.ChildForm
|
|
success_url = reverse_lazy("core:child-list")
|
|
|
|
|
|
class ChildDelete(CoreUpdateView):
|
|
model = models.Child
|
|
form_class = forms.ChildDeleteForm
|
|
template_name = "core/child_confirm_delete.html"
|
|
permission_required = ("core.delete_child",)
|
|
success_url = reverse_lazy("core:child-list")
|
|
|
|
def get_success_message(self, cleaned_data):
|
|
"""This class cannot use `CoreDeleteView` because of the confirmation
|
|
step required so the success message must be overridden."""
|
|
success_message = _("%(model)s entry deleted.") % {
|
|
"model": self.model._meta.verbose_name.title()
|
|
}
|
|
return success_message % cleaned_data
|
|
|
|
|
|
class DiaperChangeList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.DiaperChange
|
|
template_name = "core/diaperchange_list.html"
|
|
permission_required = ("core.view_diaperchange",)
|
|
paginate_by = 10
|
|
filterset_class = filters.DiaperChangeFilter
|
|
|
|
|
|
class DiaperChangeAdd(CoreAddView):
|
|
model = models.DiaperChange
|
|
permission_required = ("core.add_diaperchange",)
|
|
form_class = forms.DiaperChangeForm
|
|
success_url = reverse_lazy("core:diaperchange-list")
|
|
|
|
|
|
class DiaperChangeUpdate(CoreUpdateView):
|
|
model = models.DiaperChange
|
|
permission_required = ("core.change_diaperchange",)
|
|
form_class = forms.DiaperChangeForm
|
|
success_url = reverse_lazy("core:diaperchange-list")
|
|
|
|
|
|
class DiaperChangeDelete(CoreDeleteView):
|
|
model = models.DiaperChange
|
|
permission_required = ("core.delete_diaperchange",)
|
|
success_url = reverse_lazy("core:diaperchange-list")
|
|
|
|
|
|
class FeedingList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Feeding
|
|
template_name = "core/feeding_list.html"
|
|
permission_required = ("core.view_feeding",)
|
|
paginate_by = 10
|
|
filterset_class = filters.FeedingFilter
|
|
|
|
|
|
class FeedingAdd(CoreAddView):
|
|
model = models.Feeding
|
|
permission_required = ("core.add_feeding",)
|
|
form_class = forms.FeedingForm
|
|
success_url = reverse_lazy("core:feeding-list")
|
|
|
|
|
|
class FeedingUpdate(CoreUpdateView):
|
|
model = models.Feeding
|
|
permission_required = ("core.change_feeding",)
|
|
form_class = forms.FeedingForm
|
|
success_url = reverse_lazy("core:feeding-list")
|
|
|
|
|
|
class FeedingDelete(CoreDeleteView):
|
|
model = models.Feeding
|
|
permission_required = ("core.delete_feeding",)
|
|
success_url = reverse_lazy("core:feeding-list")
|
|
|
|
|
|
class HeadCircumferenceList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.HeadCircumference
|
|
template_name = "core/head_circumference_list.html"
|
|
permission_required = ("core.view_head_circumference",)
|
|
paginate_by = 10
|
|
filterset_class = filters.HeadCircumferenceFilter
|
|
|
|
|
|
class HeadCircumferenceAdd(CoreAddView):
|
|
model = models.HeadCircumference
|
|
template_name = "core/head_circumference_form.html"
|
|
permission_required = ("core.add_head_circumference",)
|
|
form_class = forms.HeadCircumferenceForm
|
|
success_url = reverse_lazy("core:head-circumference-list")
|
|
|
|
|
|
class HeadCircumferenceUpdate(CoreUpdateView):
|
|
model = models.HeadCircumference
|
|
template_name = "core/head_circumference_form.html"
|
|
permission_required = ("core.change_head_circumference",)
|
|
form_class = forms.HeadCircumferenceForm
|
|
success_url = reverse_lazy("core:head-circumference-list")
|
|
|
|
|
|
class HeadCircumferenceDelete(CoreDeleteView):
|
|
model = models.HeadCircumference
|
|
template_name = "core/head_circumference_confirm_delete.html"
|
|
permission_required = ("core.delete_head_circumference",)
|
|
success_url = reverse_lazy("core:head-circumference-list")
|
|
|
|
|
|
class HeightList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Height
|
|
template_name = "core/height_list.html"
|
|
permission_required = ("core.view_height",)
|
|
paginate_by = 10
|
|
filterset_class = filters.HeightFilter
|
|
|
|
|
|
class HeightAdd(CoreAddView):
|
|
model = models.Height
|
|
permission_required = ("core.add_height",)
|
|
form_class = forms.HeightForm
|
|
success_url = reverse_lazy("core:height-list")
|
|
|
|
|
|
class HeightUpdate(CoreUpdateView):
|
|
model = models.Height
|
|
permission_required = ("core.change_height",)
|
|
form_class = forms.HeightForm
|
|
success_url = reverse_lazy("core:height-list")
|
|
|
|
|
|
class HeightDelete(CoreDeleteView):
|
|
model = models.Height
|
|
permission_required = ("core.delete_height",)
|
|
success_url = reverse_lazy("core:height-list")
|
|
|
|
|
|
class NoteList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Note
|
|
template_name = "core/note_list.html"
|
|
permission_required = ("core.view_note",)
|
|
paginate_by = 10
|
|
filterset_class = filters.NoteFilter
|
|
|
|
|
|
class NoteAdd(CoreAddView):
|
|
model = models.Note
|
|
permission_required = ("core.add_note",)
|
|
form_class = forms.NoteForm
|
|
success_url = reverse_lazy("core:note-list")
|
|
|
|
|
|
class NoteUpdate(CoreUpdateView):
|
|
model = models.Note
|
|
permission_required = ("core.change_note",)
|
|
form_class = forms.NoteForm
|
|
success_url = reverse_lazy("core:note-list")
|
|
|
|
|
|
class NoteDelete(CoreDeleteView):
|
|
model = models.Note
|
|
permission_required = ("core.delete_note",)
|
|
success_url = reverse_lazy("core:note-list")
|
|
|
|
|
|
class PumpingList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Pumping
|
|
template_name = "core/pumping_list.html"
|
|
permission_required = ("core.view_pumping",)
|
|
paginate_by = 10
|
|
filterset_class = filters.PumpingFilter
|
|
|
|
|
|
class PumpingAdd(CoreAddView):
|
|
model = models.Pumping
|
|
permission_required = ("core.add_pumping",)
|
|
form_class = forms.PumpingForm
|
|
success_url = reverse_lazy("core:pumping-list")
|
|
success_message = _("%(model)s entry added!")
|
|
|
|
|
|
class PumpingUpdate(CoreUpdateView):
|
|
model = models.Pumping
|
|
permission_required = ("core.change_pumping",)
|
|
form_class = forms.PumpingForm
|
|
success_url = reverse_lazy("core:pumping-list")
|
|
success_message = _("%(model)s entry for %(child)s updated.")
|
|
|
|
|
|
class PumpingDelete(CoreDeleteView):
|
|
model = models.Pumping
|
|
permission_required = ("core.delete_pumping",)
|
|
success_url = reverse_lazy("core:pumping-list")
|
|
|
|
|
|
class SleepList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Sleep
|
|
template_name = "core/sleep_list.html"
|
|
permission_required = ("core.view_sleep",)
|
|
paginate_by = 10
|
|
filterset_class = filters.SleepFilter
|
|
|
|
|
|
class SleepAdd(CoreAddView):
|
|
model = models.Sleep
|
|
permission_required = ("core.add_sleep",)
|
|
form_class = forms.SleepForm
|
|
success_url = reverse_lazy("core:sleep-list")
|
|
|
|
|
|
class SleepUpdate(CoreUpdateView):
|
|
model = models.Sleep
|
|
permission_required = ("core.change_sleep",)
|
|
form_class = forms.SleepForm
|
|
success_url = reverse_lazy("core:sleep-list")
|
|
|
|
|
|
class SleepDelete(CoreDeleteView):
|
|
model = models.Sleep
|
|
permission_required = ("core.delete_sleep",)
|
|
success_url = reverse_lazy("core:sleep-list")
|
|
|
|
|
|
class TemperatureList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Temperature
|
|
template_name = "core/temperature_list.html"
|
|
permission_required = ("core.view_temperature",)
|
|
paginate_by = 10
|
|
filterset_class = filters.TemperatureFilter
|
|
|
|
|
|
class TemperatureAdd(CoreAddView):
|
|
model = models.Temperature
|
|
permission_required = ("core.add_temperature",)
|
|
form_class = forms.TemperatureForm
|
|
success_url = reverse_lazy("core:temperature-list")
|
|
success_message = _("%(model)s reading added!")
|
|
|
|
|
|
class TemperatureUpdate(CoreUpdateView):
|
|
model = models.Temperature
|
|
permission_required = ("core.change_temperature",)
|
|
form_class = forms.TemperatureForm
|
|
success_url = reverse_lazy("core:temperature-list")
|
|
success_message = _("%(model)s reading for %(child)s updated.")
|
|
|
|
|
|
class TemperatureDelete(CoreDeleteView):
|
|
model = models.Temperature
|
|
permission_required = ("core.delete_temperature",)
|
|
success_url = reverse_lazy("core:temperature-list")
|
|
|
|
|
|
class Timeline(LoginRequiredMixin, TemplateView):
|
|
template_name = "timeline/timeline.html"
|
|
|
|
# Show the overall timeline or a child timeline if one Child instance.
|
|
def get(self, request, *args, **kwargs):
|
|
children = models.Child.objects.count()
|
|
if children == 1:
|
|
return HttpResponseRedirect(
|
|
reverse("core:child", args={models.Child.objects.first().slug})
|
|
)
|
|
return super(Timeline, self).get(request, *args, **kwargs)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(Timeline, self).get_context_data(**kwargs)
|
|
date = self.request.GET.get("date", str(timezone.localdate()))
|
|
_prepare_timeline_context_data(context, date)
|
|
return context
|
|
|
|
|
|
class TimerList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Timer
|
|
template_name = "core/timer_list.html"
|
|
permission_required = ("core.view_timer",)
|
|
paginate_by = 10
|
|
filterset_fields = ("user",)
|
|
|
|
|
|
class TimerDetail(PermissionRequiredMixin, DetailView):
|
|
model = models.Timer
|
|
permission_required = ("core.view_timer",)
|
|
|
|
|
|
class TimerAdd(PermissionRequiredMixin, CreateView):
|
|
model = models.Timer
|
|
permission_required = ("core.add_timer",)
|
|
form_class = forms.TimerForm
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super(TimerAdd, self).get_form_kwargs()
|
|
kwargs.update({"user": self.request.user})
|
|
return kwargs
|
|
|
|
def get_success_url(self):
|
|
return reverse("core:timer-detail", kwargs={"pk": self.object.pk})
|
|
|
|
|
|
class TimerUpdate(CoreUpdateView):
|
|
model = models.Timer
|
|
permission_required = ("core.change_timer",)
|
|
form_class = forms.TimerForm
|
|
success_url = reverse_lazy("core:timer-list")
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super(TimerUpdate, self).get_form_kwargs()
|
|
kwargs.update({"user": self.request.user})
|
|
return kwargs
|
|
|
|
def get_success_url(self):
|
|
instance = self.get_object()
|
|
return reverse("core:timer-detail", kwargs={"pk": instance.pk})
|
|
|
|
|
|
class TimerAddQuick(PermissionRequiredMixin, RedirectView):
|
|
http_method_names = ["post"]
|
|
permission_required = ("core.add_timer",)
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
instance = models.Timer.objects.create(user=request.user)
|
|
# Find child from child pk in POST
|
|
child_id = request.POST.get("child", False)
|
|
child = models.Child.objects.get(pk=child_id) if child_id else None
|
|
if child:
|
|
instance.child = child
|
|
# Add child relationship if there is only Child instance.
|
|
elif models.Child.count() == 1:
|
|
instance.child = models.Child.objects.first()
|
|
instance.save()
|
|
self.url = request.GET.get(
|
|
"next", reverse("core:timer-detail", args={instance.id})
|
|
)
|
|
return super(TimerAddQuick, self).get(request, *args, **kwargs)
|
|
|
|
|
|
class TimerRestart(PermissionRequiredMixin, RedirectView):
|
|
http_method_names = ["post"]
|
|
permission_required = ("core.change_timer",)
|
|
|
|
def post(self, request, *args, **kwargs):
|
|
instance = models.Timer.objects.get(id=kwargs["pk"])
|
|
instance.restart()
|
|
messages.success(request, "{} restarted.".format(instance))
|
|
return super(TimerRestart, self).get(request, *args, **kwargs)
|
|
|
|
def get_redirect_url(self, *args, **kwargs):
|
|
return reverse("core:timer-detail", kwargs={"pk": kwargs["pk"]})
|
|
|
|
|
|
class TimerDelete(CoreDeleteView):
|
|
model = models.Timer
|
|
permission_required = ("core.delete_timer",)
|
|
success_url = reverse_lazy("core:timer-list")
|
|
|
|
|
|
class TummyTimeList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.TummyTime
|
|
template_name = "core/tummytime_list.html"
|
|
permission_required = ("core.view_tummytime",)
|
|
paginate_by = 10
|
|
filterset_class = filters.TummyTimeFilter
|
|
|
|
|
|
class TummyTimeAdd(CoreAddView):
|
|
model = models.TummyTime
|
|
permission_required = ("core.add_tummytime",)
|
|
form_class = forms.TummyTimeForm
|
|
success_url = reverse_lazy("core:tummytime-list")
|
|
|
|
|
|
class TummyTimeUpdate(CoreUpdateView):
|
|
model = models.TummyTime
|
|
permission_required = ("core.change_tummytime",)
|
|
form_class = forms.TummyTimeForm
|
|
success_url = reverse_lazy("core:tummytime-list")
|
|
|
|
|
|
class TummyTimeDelete(CoreDeleteView):
|
|
model = models.TummyTime
|
|
permission_required = ("core.delete_tummytime",)
|
|
success_url = reverse_lazy("core:tummytime-list")
|
|
|
|
|
|
class WeightList(PermissionRequiredMixin, BabyBuddyFilterView):
|
|
model = models.Weight
|
|
template_name = "core/weight_list.html"
|
|
permission_required = ("core.view_weight",)
|
|
paginate_by = 10
|
|
filterset_class = filters.WeightFilter
|
|
|
|
|
|
class WeightAdd(CoreAddView):
|
|
model = models.Weight
|
|
permission_required = ("core.add_weight",)
|
|
form_class = forms.WeightForm
|
|
success_url = reverse_lazy("core:weight-list")
|
|
|
|
|
|
class WeightUpdate(CoreUpdateView):
|
|
model = models.Weight
|
|
permission_required = ("core.change_weight",)
|
|
form_class = forms.WeightForm
|
|
success_url = reverse_lazy("core:weight-list")
|
|
|
|
|
|
class WeightDelete(CoreDeleteView):
|
|
model = models.Weight
|
|
permission_required = ("core.delete_weight",)
|
|
success_url = reverse_lazy("core:weight-list")
|