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

Add a "days" argument to the fake command for number of days of data to fake for each child.

This commit is contained in:
Christopher Charbonneau Wells 2017-08-21 19:34:06 -04:00
parent 8ae33b0f48
commit 51946dd28a
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
babyblotter/management/commands/fake.py
View File

@ -27,10 +27,17 @@ class Command(BaseCommand):
default=1, default=1,
help='The number of fake children to create.' help='The number of fake children to create.'
) )
parser.add_argument(
'--days',
dest='days',
default=5,
help='How many days of fake entries to create.'
)
def handle(self, *args, **kwargs): def handle(self, *args, **kwargs):
verbosity = kwargs['verbosity'] or 1 verbosity = int(kwargs['verbosity']) or 1
children = kwargs['children'] or 1 children = int(kwargs['children']) or 1
days = int(kwargs['days']) or 5
for i in range(0, children): for i in range(0, children):
child = Child.objects.create( child = Child.objects.create(
@ -40,8 +47,9 @@ class Command(BaseCommand):
) )
child.save() child.save()
for j in range(1, 6): for j in range(0, days):
date = timezone.now() - timedelta(days=j) date = (timezone.now() - timedelta(days=j)).replace(
hour=0, minute=0, second=0)
self._add_child_data(child, date) self._add_child_data(child, date)
if verbosity > 0: if verbosity > 0:
@ -62,14 +70,14 @@ class Command(BaseCommand):
DiaperChange.objects.create( DiaperChange.objects.create(
child=child, child=child,
time=date + timedelta(seconds=randint(-86400, 86400)), time=date + timedelta(seconds=randint(0, 86400)),
wet=wet, wet=wet,
solid=solid, solid=solid,
color=color color=color
).save() ).save()
for i in (range(0, randint(5, 20))): for i in (range(0, randint(5, 20))):
start = date + timedelta(seconds=randint(-86400, 86400)) start = date + timedelta(seconds=randint(0, 86400))
method = choice(Feeding._meta.get_field('method').choices)[0] method = choice(Feeding._meta.get_field('method').choices)[0]
if method is 'bottle': if method is 'bottle':
@ -87,7 +95,7 @@ class Command(BaseCommand):
).save() ).save()
for i in (range(0, randint(2, 10))): for i in (range(0, randint(2, 10))):
start = date + timedelta(seconds=randint(-86400, 86400)) start = date + timedelta(seconds=randint(0, 86400))
Sleep.objects.create( Sleep.objects.create(
child=child, child=child,
@ -96,7 +104,7 @@ class Command(BaseCommand):
).save() ).save()
for i in (range(0, randint(2, 10))): for i in (range(0, randint(2, 10))):
start = date + timedelta(seconds=randint(-86400, 86400)) start = date + timedelta(seconds=randint(0, 86400))
if choice([True, False]): if choice([True, False]):
milestone = self.faker.sentence() milestone = self.faker.sentence()