diff options
Diffstat (limited to 'app/dispatch/management/commands')
| -rw-r--r-- | app/dispatch/management/commands/insert_fake_data.py | 11 | ||||
| -rw-r--r-- | app/dispatch/management/commands/setup.py | 15 |
2 files changed, 25 insertions, 1 deletions
diff --git a/app/dispatch/management/commands/insert_fake_data.py b/app/dispatch/management/commands/insert_fake_data.py index 5d4203d..2969900 100644 --- a/app/dispatch/management/commands/insert_fake_data.py +++ b/app/dispatch/management/commands/insert_fake_data.py @@ -1,6 +1,6 @@ from django.core.management.base import BaseCommand, CommandError from django.contrib.auth import get_user_model -from dispatch.models import Customer, Load +from dispatch.models import Customer, Load, Identity from faker import Faker import random @@ -106,6 +106,15 @@ class Command(BaseCommand): is_active=True ) new_user.save() + user_ident = Identity( + user=new_user, + name="{} {}".format(ffname, flname), + address=self.fake.address(), + city=self.fake.city(), + state=self.fake.state(), + zip_code=self.fake.zipcode(), + ) + user_ident.save() return new_user except IntegrityError: # Around and around we go until we get something new diff --git a/app/dispatch/management/commands/setup.py b/app/dispatch/management/commands/setup.py new file mode 100644 index 0000000..bdfa96e --- /dev/null +++ b/app/dispatch/management/commands/setup.py @@ -0,0 +1,15 @@ +from django.core.management.base import BaseCommand, CommandError +from django.contrib.auth import get_user_model +from dispatch.models import Customer, Load +from faker import Faker +import random + +from django.db.utils import IntegrityError + +# import yaml + +User = get_user_model() + + +class Command(BaseCommand): + help = "Run with " |
