Make sure there is at least one Feeding object before attempting to p…

…reset the type for a new entry.
babybuddy · Apr 7, 2018 · 7c98015 · 7c98015
1 parent 382d106
commit 7c98015
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/forms.py b/core/forms.py
@@ -41,7 +41,7 @@ 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.latest('end'):
+    if instance is None and models.Feeding.objects.count() > 0:
         kwargs['initial'].update({
             'type': models.Feeding.objects.latest('end').type
         })