1 files changed, 17 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'])
+