mybuddy/core/forms.py

243 lines
7.4 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
from django import forms
from django.conf import settings
from django.utils import timezone
from django.utils.translation import gettext as _
2017-11-10 00:50:54 +00:00
from core import models
# Sets the default Child instance if only one exists in the database.
def set_default_child(kwargs):
instance = kwargs.get('instance', None)
if not kwargs.get('initial'):
kwargs.update(initial={})
2017-11-10 00:50:54 +00:00
if instance is None and models.Child.objects.count() == 1:
kwargs['initial'].update({'child': models.Child.objects.first()})
return kwargs
# Uses a timer to set the default start and end date and updates the timer.
def set_default_duration(kwargs):
instance = kwargs.get('instance', None)
timer_id = kwargs.get('timer', None)
if not kwargs.get('initial'):
kwargs.update(initial={})
if not instance and timer_id:
2017-11-10 00:50:54 +00:00
instance = models.Timer.objects.get(id=timer_id)
kwargs['initial'].update({
'timer': instance,
'start': instance.start,
'end': instance.end or timezone.now()
})
try:
kwargs.pop('timer')
except KeyError:
pass
return kwargs
# Sets the default Feeding type to the one used in the most recent entry.
def set_default_feeding_type(kwargs):
instance = kwargs.get('instance', None)
if not kwargs.get('initial'):
kwargs.update(initial={})
if instance is None and models.Feeding.objects.count() > 0:
kwargs['initial'].update({
'type': models.Feeding.objects.latest('end').type
})
return kwargs
class ChildForm(forms.ModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.Child
fields = [
'first_name',
'last_name',
'birth_date'
]
2017-12-01 23:07:08 +00:00
if settings.BABY_BUDDY['ALLOW_UPLOADS']:
fields.append('picture')
widgets = {
'birth_date': forms.DateInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_date',
}),
}
class ChildDeleteForm(forms.ModelForm):
confirm_name = forms.CharField(max_length=511)
class Meta:
2017-11-10 00:50:54 +00:00
model = models.Child
fields = []
def clean_confirm_name(self):
confirm_name = self.cleaned_data['confirm_name']
if confirm_name != str(self.instance):
2017-11-01 20:14:42 +00:00
raise forms.ValidationError(
_('Name does not match child name.'), code='confirm_mismatch')
return confirm_name
def save(self, commit=True):
instance = self.instance
self.instance.delete()
return instance
class DiaperChangeForm(forms.ModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.DiaperChange
fields = ['child', 'time', 'wet', 'solid', 'color']
widgets = {
'time': forms.DateTimeInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_time',
}),
}
def __init__(self, *args, **kwargs):
kwargs = set_default_child(kwargs)
super(DiaperChangeForm, self).__init__(*args, **kwargs)
class FeedingForm(forms.ModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.Feeding
fields = ['child', 'start', 'end', 'type', 'method', 'amount']
widgets = {
'start': forms.DateTimeInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_start',
}),
'end': forms.DateTimeInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_end',
}),
}
def __init__(self, *args, **kwargs):
kwargs = set_default_child(kwargs)
kwargs = set_default_feeding_type(kwargs)
self.timer_id = kwargs.get('timer', None)
kwargs = set_default_duration(kwargs)
super(FeedingForm, self).__init__(*args, **kwargs)
def save(self, commit=True):
instance = super(FeedingForm, self).save(commit=False)
if self.timer_id:
2017-11-10 00:50:54 +00:00
timer = models.Timer.objects.get(id=self.timer_id)
timer.stop(instance.end)
instance.save()
return instance
2017-11-10 00:50:54 +00:00
class NoteForm(forms.ModelForm):
class Meta:
model = models.Note
fields = ['child', 'note']
def __init__(self, *args, **kwargs):
kwargs = set_default_child(kwargs)
super(NoteForm, self).__init__(*args, **kwargs)
class SleepForm(forms.ModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.Sleep
fields = ['child', 'start', 'end']
widgets = {
'start': forms.DateTimeInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_start',
}),
'end': forms.DateTimeInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_end',
}),
}
def __init__(self, *args, **kwargs):
kwargs = set_default_child(kwargs)
self.timer_id = kwargs.get('timer', None)
kwargs = set_default_duration(kwargs)
super(SleepForm, self).__init__(*args, **kwargs)
def save(self, commit=True):
instance = super(SleepForm, self).save(commit=False)
if self.timer_id:
2017-11-10 00:50:54 +00:00
timer = models.Timer.objects.get(id=self.timer_id)
timer.stop(instance.end)
instance.save()
return instance
2017-08-17 01:57:01 +00:00
class TimerForm(forms.ModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.Timer
fields = ['name', 'start']
widgets = {
'start': forms.DateTimeInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_start',
})
}
2017-08-17 01:57:01 +00:00
def __init__(self, *args, **kwargs):
2017-09-10 13:50:16 +00:00
self.user = kwargs.pop('user')
super(TimerForm, self).__init__(*args, **kwargs)
def save(self, commit=True):
instance = super(TimerForm, self).save(commit=False)
2017-09-10 13:50:16 +00:00
instance.user = self.user
instance.save()
return instance
2017-08-17 01:57:01 +00:00
class TummyTimeForm(forms.ModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.TummyTime
fields = ['child', 'start', 'end', 'milestone']
widgets = {
'start': forms.DateTimeInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_start',
}),
'end': forms.DateTimeInput(attrs={
'class': 'datetimepicker-input',
'data-target': '#datetimepicker_end',
}),
}
def __init__(self, *args, **kwargs):
kwargs = set_default_child(kwargs)
self.timer_id = kwargs.get('timer', None)
kwargs = set_default_duration(kwargs)
super(TummyTimeForm, self).__init__(*args, **kwargs)
def save(self, commit=True):
instance = super(TummyTimeForm, self).save(commit=False)
if self.timer_id:
2017-11-10 00:50:54 +00:00
timer = models.Timer.objects.get(id=self.timer_id)
timer.stop(instance.end)
instance.save()
return instance
2017-11-10 02:15:09 +00:00
class WeightForm(forms.ModelForm):
class Meta:
model = models.Weight
fields = ['child', 'weight', 'date']
widgets = {
'date': forms.DateInput(attrs={
'class': 'datetimepicker-input',
2017-11-10 02:15:09 +00:00
'data-target': '#datetimepicker_date',
}),
}
def __init__(self, *args, **kwargs):
kwargs = set_default_child(kwargs)
super(WeightForm, self).__init__(*args, **kwargs)