mybuddy/core/forms.py

283 lines
8.6 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
def set_initial_values(kwargs, form_type):
"""
Sets initial value for add forms based on provided kwargs.
:param kwargs: Keyword arguments.
:param form_type: Class of the type of form being initialized.
:return: Keyword arguments with updated "initial" values.
"""
# Never update initial values for existing instance (e.g. edit operation).
if kwargs.get('instance', None):
return kwargs
# Add the "initial" kwarg if it does not already exist.
if not kwargs.get('initial'):
kwargs.update(initial={})
# Set Child based on `child` kwarg or single Chile database.
child_slug = kwargs.get('child', None)
if child_slug:
kwargs['initial'].update({
'child': models.Child.objects.filter(slug=child_slug).first(),
})
elif models.Child.count() == 1:
kwargs['initial'].update({'child': models.Child.objects.first()})
# Set start and end time based on Timer from `timer` kwarg.
timer_id = kwargs.get('timer', None)
if timer_id:
timer = models.Timer.objects.get(id=timer_id)
kwargs['initial'].update({
'timer': timer,
'start': timer.start,
'end': timer.end or timezone.now()
})
# Set type and method values for Feeding instance based on last feed.
if form_type == FeedingForm and 'child' in kwargs['initial']:
last_feeding = models.Feeding.objects.filter(
child=kwargs['initial']['child']).order_by('end').last()
if last_feeding:
last_method = last_feeding.method
last_feed_args = {'type': last_feeding.type}
if last_method not in ['left breast', 'right breast']:
last_feed_args['method'] = last_method
kwargs['initial'].update(last_feed_args)
# Remove custom kwargs so they do not interfere with `super` calls.
for key in ['child', 'timer']:
try:
kwargs.pop(key)
except KeyError:
pass
return kwargs
class CoreModelForm(forms.ModelForm):
def __init__(self, *args, **kwargs):
# Set `timer_id` so the Timer can be stopped in the `save` method.
self.timer_id = kwargs.get('timer', None)
kwargs = set_initial_values(kwargs, type(self))
super(CoreModelForm, self).__init__(*args, **kwargs)
def save(self, commit=True):
# If `timer_id` is present, stop the Timer.
instance = super(CoreModelForm, self).save(commit=False)
if self.timer_id:
timer = models.Timer.objects.get(id=self.timer_id)
timer.stop(instance.end)
if commit:
instance.save()
return instance
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={
'autocomplete': 'off',
'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(CoreModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.DiaperChange
2020-02-15 04:10:08 +00:00
fields = ['child', 'time', 'wet', 'solid', 'color', 'amount', 'notes']
widgets = {
'time': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_time',
}),
2020-02-16 22:03:50 +00:00
'notes': forms.Textarea(attrs={'rows': 5}),
}
class FeedingForm(CoreModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.Feeding
2020-02-15 04:10:08 +00:00
fields = ['child', 'start', 'end', 'type', 'method', 'amount', 'notes']
widgets = {
'start': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_start',
}),
'end': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_end',
}),
2020-02-16 22:03:50 +00:00
'notes': forms.Textarea(attrs={'rows': 5}),
}
class NoteForm(CoreModelForm):
2017-11-10 00:50:54 +00:00
class Meta:
model = models.Note
fields = ['child', 'note', 'time']
widgets = {
'time': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_time',
}),
}
2017-11-10 00:50:54 +00:00
class SleepForm(CoreModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.Sleep
2020-02-15 04:10:08 +00:00
fields = ['child', 'start', 'end', 'notes']
widgets = {
'start': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_start',
}),
'end': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_end',
}),
2020-02-16 22:03:50 +00:00
'notes': forms.Textarea(attrs={'rows': 5}),
}
class TemperatureForm(CoreModelForm):
2019-05-17 04:33:26 +00:00
class Meta:
model = models.Temperature
2020-02-15 04:10:08 +00:00
fields = ['child', 'temperature', 'time', 'notes']
2019-05-17 04:33:26 +00:00
widgets = {
'time': forms.DateTimeInput(attrs={
'autocomplete': 'off',
2019-05-17 04:33:26 +00:00
'data-target': '#datetimepicker_time',
}),
2020-02-16 22:03:50 +00:00
'notes': forms.Textarea(attrs={'rows': 5}),
2019-05-17 04:33:26 +00:00
}
class TimerForm(CoreModelForm):
2017-08-17 01:57:01 +00:00
class Meta:
2017-11-10 00:50:54 +00:00
model = models.Timer
2020-01-28 21:56:28 +00:00
fields = ['child', 'name', 'start']
widgets = {
'start': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'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(CoreModelForm):
class Meta:
2017-11-10 00:50:54 +00:00
model = models.TummyTime
fields = ['child', 'start', 'end', 'milestone']
widgets = {
'start': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_start',
}),
'end': forms.DateTimeInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_end',
}),
}
2017-11-10 02:15:09 +00:00
class WeightForm(CoreModelForm):
2017-11-10 02:15:09 +00:00
class Meta:
model = models.Weight
2020-02-15 04:10:08 +00:00
fields = ['child', 'weight', 'date', 'notes']
2017-11-10 02:15:09 +00:00
widgets = {
'date': forms.DateInput(attrs={
'autocomplete': 'off',
2017-11-10 02:15:09 +00:00
'data-target': '#datetimepicker_date',
}),
2020-02-16 22:03:50 +00:00
'notes': forms.Textarea(attrs={'rows': 5}),
2017-11-10 02:15:09 +00:00
}
2021-12-30 03:32:55 +00:00
class HeightForm(CoreModelForm):
class Meta:
model = models.Height
fields = ['child', 'height', 'date', 'notes']
widgets = {
'date': forms.DateInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_date',
}),
'notes': forms.Textarea(attrs={'rows': 5}),
}
2021-12-30 03:32:55 +00:00
class HeadCircumferenceForm(CoreModelForm):
class Meta:
model = models.HeadCircumference
fields = ['child', 'head_circumference', 'date', 'notes']
widgets = {
'date': forms.DateInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_date',
}),
'notes': forms.Textarea(attrs={'rows': 5}),
}
2021-12-30 03:32:55 +00:00
class BMIForm(CoreModelForm):
class Meta:
model = models.BMI
fields = ['child', 'bmi', 'date', 'notes']
widgets = {
'date': forms.DateInput(attrs={
'autocomplete': 'off',
'data-target': '#datetimepicker_date',
}),
'notes': forms.Textarea(attrs={'rows': 5}),
2021-12-30 03:32:55 +00:00
}