From 7918b9aca8dc1f159f6f57f2bac815e814fc4973 Mon Sep 17 00:00:00 2001 From: John Merlie Date: Fri, 16 Jun 2023 17:12:18 -0400 Subject: [PATCH] updated pumping fake data for start/end/duration --- babybuddy/management/commands/fake.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/babybuddy/management/commands/fake.py b/babybuddy/management/commands/fake.py index 288a04ae..f7ec8f75 100644 --- a/babybuddy/management/commands/fake.py +++ b/babybuddy/management/commands/fake.py @@ -145,9 +145,13 @@ class Command(BaseCommand): if choice([True, False, False, False]): notes = " ".join(self.faker.sentences(randint(1, 5))) - models.Pumping.objects.create( - child=self.child, amount=self.amount, time=self.time, notes=notes - ).save() + start = self.time + timedelta(minutes=randint(1, 60)) + end = start + timedelta(minutes=randint(5, 20)) + + if end < self.time_now: + models.Pumping.objects.create( + child=self.child, amount=self.amount, start=start, end=end, notes=notes + ).save() @transaction.atomic def _add_diaperchange_entry(self):