steve
/
mybuddy
mirror of https://github.com/snachodog/mybuddy.git
1
0
Fork 0
Code Issues 3 Packages Projects 1 Releases Wiki Activity

Prepopulate method for formula types (#127).

This commit is contained in:
Peter Hardy 2020-03-26 22:15:08 +11:00
parent 4d74df27d9
commit 4978208d7f
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
core/forms.py
View File

@ -43,12 +43,16 @@ def set_initial_values(kwargs, form_type):
'end': timer.end or timezone.now() 'end': timer.end or timezone.now()
}) })
# Set initial type value for Feeding instance based on last type used. # Set type and method values for Feeding instance based on last feed.
if form_type == FeedingForm and 'child' in kwargs['initial']: if form_type == FeedingForm and 'child' in kwargs['initial']:
last_feeding = models.Feeding.objects.filter( last_feeding = models.Feeding.objects.filter(
child=kwargs['initial']['child']).order_by('end').last() child=kwargs['initial']['child']).order_by('end').last()
if last_feeding: if last_feeding:
kwargs['initial'].update({'type': last_feeding.type}) last_type = last_feeding.type
last_feed_args = {'type': last_feeding.type}
if last_type == 'formula':
last_feed_args['method'] = 'bottle'
kwargs['initial'].update(last_feed_args)
# Remove custom kwargs so they do not interfere with `super` calls. # Remove custom kwargs so they do not interfere with `super` calls.
for key in ['child', 'timer']: for key in ['child', 'timer']: