2 files changed, 54 insertions, 2 deletions
diff --git a/app/dispatch/management/commands/setup.py b/app/dispatch/management/commands/setup.py
index bdfa96e..8c72a94 100644
--- a/app/dispatch/management/commands/setup.py
+++ b/app/dispatch/management/commands/setup.py
@@ -1,6 +1,5 @@
 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
 
@@ -12,4 +11,20 @@ User = get_user_model()
 
 
 class Command(BaseCommand):
-    help = "Run with "
+    help = "Sets up the default Super User for the application, as a demo"
+
+    def add_arguments(self, parser):
+        # parser.add_argument('--file', type=str, dest='filename')
+        parser.add_argument('--fname', type=str, dest='fname', default='Admin')
+        parser.add_argument('--lname', type=str, dest='lname', default='User')
+        parser.add_argument('--email', type=str, dest='email', default='admin@example.com')
+        parser.add_argument('--password', type=str, dest='password', default='password')
+
+
+    def handle(self, *args, **options):
+        User.objects.create_superuser(
+                email=options['email'],
+                first_name=options['fname'],
+                last_name=options['lname'],
+                password=options['password'])
+
diff --git a/app/dispatch/management/commands/setup_identity.py b/app/dispatch/management/commands/setup_identity.py
new file mode 100644
index 0000000..21ad59c
--- /dev/null
+++ b/app/dispatch/management/commands/setup_identity.py
@@ -0,0 +1,37 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.contrib.auth import get_user_model
+from dispatch.models import Identity
+from faker import Faker
+import random
+
+from django.db.utils import IntegrityError
+
+# import yaml
+
+User = get_user_model()
+
+
+class Command(BaseCommand):
+    help = "Sets up the default Identity for the application, --name $name --address $addres --city $city --state $state --zip $zip"
+
+    def add_arguments(self, parser):
+        # parser.add_argument('--file', type=str, dest='filename')
+        parser.add_argument('--name', type=str, dest='name', default='My Company')
+        parser.add_argument('--address', type=str, dest='address', default='123 Main St')
+        parser.add_argument('--city', type=str, dest='city', default='Demoville')
+        parser.add_argument('--state', type=str, dest='state', default='Michigan')
+        parser.add_argument('--zip', type=str, dest='zip', default='49503')
+
+
+    def handle(self, *args, **options):
+        u = User.objects.first()
+
+        ident = Identity(
+                user=u,
+                name=options['name'],
+                address=options['address'],
+                city=options['city'],
+                state=options['state'],
+                zip_code=options['zip'])
+        ident.save()
+        ident.set_default()