# -*- coding: utf-8 -*- from random import choice, randint, uniform from datetime import timedelta from decimal import Decimal from django.db import transaction from django.core.management.base import BaseCommand from django.utils import timezone from faker import Factory from core import models class Command(BaseCommand): help = 'Generates fake children and related entries.' def __init__(self, *args, **kwargs): super(Command, self).__init__(*args, **kwargs) self.faker = Factory.create() self.child = None self.weight = None self.time = None self.time_now = timezone.localtime() def add_arguments(self, parser): parser.add_argument( '--children', dest='children', default=1, help='The number of fake children to create.' ) parser.add_argument( '--days', dest='days', default=31, help='How many days of fake entries to create.' ) def handle(self, *args, **kwargs): verbosity = int(kwargs['verbosity']) children = int(kwargs['children']) or 1 days = int(kwargs['days']) or 31 birth_date = (timezone.localtime() - timedelta(days=days)) for i in range(0, children): self.child = models.Child.objects.create( first_name=self.faker.first_name(), last_name=self.faker.last_name(), birth_date=birth_date ) self.child.save() self._add_child_data() if verbosity > 0: self.stdout.write( self.style.SUCCESS('Successfully added fake data.') ) @transaction.atomic def _add_child_data(self): """ Adds fake data for child from child.birth_date to now. The fake data follows a semi-regular pattern of sleep, feed, change, tummy time, change, tummy time, sleep, etc. :returns: """ self.time = self.child.birth_date self.temperature = round(uniform(95.0, 102.0), 2) self._add_temperature_entry() self.weight = round(uniform(8.0, 12.0), 2) self._add_weight_entry() last_weight_entry_time = self.time self._add_note_entry() while self.time < self.time_now: self._add_sleep_entry() if choice([True, False]): self._add_diaperchange_entry() self._add_feeding_entry() self._add_diaperchange_entry() if choice([True, False]): self._add_tummytime_entry() if choice([True, False]): self._add_diaperchange_entry() self._add_tummytime_entry() if choice([True, False]): self._add_temperature_entry() if (self.time - last_weight_entry_time).days > 6: self._add_weight_entry() last_weight_entry_time = self.time @transaction.atomic def _add_diaperchange_entry(self): """ Add a Diaper Change entry and advance self.time. :returns: """ solid = choice([True, False, False, False]) wet = choice([True, False]) color = '' if solid: color = choice( models.DiaperChange._meta.get_field('color').choices)[0] if not wet and not solid: wet = True amount = Decimal('%d.%d' % (randint(0, 6), randint(1, 9))) time = self.time + timedelta(minutes=randint(1, 60)) notes = '' if choice([True, False, False, False]): notes = ' '.join(self.faker.sentences(randint(1, 5))) if time < self.time_now: models.DiaperChange.objects.create( child=self.child, time=time, wet=wet, solid=solid, color=color, amount=amount, notes=notes ).save() self.time = time @transaction.atomic def _add_feeding_entry(self): """ Add a Feeding entry and advance self.time. :returns: """ method = choice(models.Feeding._meta.get_field('method').choices)[0] amount = None if method == 'bottle': amount = Decimal('%d.%d' % (randint(0, 6), randint(0, 9))) start = self.time + timedelta(minutes=randint(1, 60)) end = start + timedelta(minutes=randint(5, 20)) notes = '' if choice([True, False, False, False]): notes = ' '.join(self.faker.sentences(randint(1, 5))) if end < self.time_now: models.Feeding.objects.create( child=self.child, start=start, end=end, type=choice(models.Feeding._meta.get_field('type').choices)[0], method=method, amount=amount, notes=notes ).save() self.time = end @transaction.atomic def _add_note_entry(self): """ Add a Note entry. :returns: """ note = self.faker.sentence() models.Note.objects.create(child=self.child, note=note).save() @transaction.atomic def _add_sleep_entry(self): """ Add a Sleep entry and advance self.time. Between the hours of 18 and 6, extend the minimum and maximum sleep duration settings. :returns: """ if self.time.hour < 6 or self.time.hour > 18: minutes = randint(60 * 2, 60 * 6) else: minutes = randint(30, 60 * 2) end = self.time + timedelta(minutes=minutes) notes = '' if choice([True, False, False, False]): notes = ' '.join(self.faker.sentences(randint(1, 5))) if end < self.time_now: models.Sleep.objects.create( child=self.child, start=self.time, end=end, notes=notes ).save() self.time = end @transaction.atomic def _add_temperature_entry(self): """ Add a Temperature entry. This assumes a weekly interval. :returns: """ self.temperature = round(uniform(95.0, 102.0), 2) notes = '' if choice([True, False, False, False]): notes = ' '.join(self.faker.sentences(randint(1, 5))) models.Temperature.objects.create( child=self.child, temperature=self.temperature, time=self.time, notes=notes ).save() @transaction.atomic def _add_tummytime_entry(self): """ Add a Tummy time entry and advance self.time. :returns: """ milestone = '' if choice([True, False]): milestone = self.faker.sentence() start = self.time + timedelta(minutes=randint(1, 60)) end = start + timedelta(minutes=randint(0, 10), seconds=randint(0, 59)) if (end - start).seconds < 20: end = start + timedelta(minutes=1, seconds=30) if end < self.time_now: models.TummyTime.objects.create( child=self.child, start=start, end=end, milestone=milestone ).save() self.time = end @transaction.atomic def _add_weight_entry(self): """ Add a Weight entry. This assumes a weekly interval. :returns: """ self.weight += uniform(0.1, 0.3) notes = '' if choice([True, False, False, False]): notes = ' '.join(self.faker.sentences(randint(1, 5))) models.Weight.objects.create( child=self.child, weight=round(self.weight, 2), date=self.time.date(), notes=notes ).save()