aboutsummaryrefslogtreecommitdiff
path: root/app/dispatch/management
diff options
context:
space:
mode:
Diffstat (limited to 'app/dispatch/management')
-rw-r--r--app/dispatch/management/commands/insert_fake_data.py11
-rw-r--r--app/dispatch/management/commands/setup.py15
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 "