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,
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):
verbosity = kwargs['verbosity'] or 1
children = kwargs['children'] or 1
verbosity = int(kwargs['verbosity']) or 1
children = int(kwargs['children']) or 1
days = int(kwargs['days']) or 5
for i in range(0, children):
child = Child.objects.create(
@ -40,8 +47,9 @@ class Command(BaseCommand):
)
child.save()
for j in range(1, 6):
date = timezone.now() - timedelta(days=j)
for j in range(0, days):
date = (timezone.now() - timedelta(days=j)).replace(
hour=0, minute=0, second=0)
self._add_child_data(child, date)
if verbosity > 0:
@ -62,14 +70,14 @@ class Command(BaseCommand):
DiaperChange.objects.create(
child=child,
time=date + timedelta(seconds=randint(-86400, 86400)),
time=date + timedelta(seconds=randint(0, 86400)),
wet=wet,
solid=solid,
color=color
).save()
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]
if method is 'bottle':
@ -87,7 +95,7 @@ class Command(BaseCommand):
).save()
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(
child=child,
@ -96,7 +104,7 @@ class Command(BaseCommand):
).save()
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]):
milestone = self.faker.sentence()