Re-add random color for solid Diaper Changes in `fake` command.

2017-11-18 06:55:39 -05:00 · 2017-11-18 06:55:39 -05:00 · 4aa51e44da
parent eff1320a5e
commit 4aa51e44da
1 changed files with 3 additions and 0 deletions
--- a/babybuddy/management/commands/fake.py
+++ b/babybuddy/management/commands/fake.py
@ -96,6 +96,9 @@ class Command(BaseCommand):
        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
        time = self.time + timedelta(minutes=randint(1, 60))