From 0e82caee0b427dcb957ff78a6c2779e7d9e27dde Mon Sep 17 00:00:00 2001 From: Mitch Riedstra Date: Sat, 14 Sep 2019 00:06:50 -0400 Subject: Fully automated setup and forward port 36200 --- app/dispatch/management/commands/setup.py | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) (limited to 'app/dispatch/management/commands/setup.py') 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']) + -- cgit v1.2.3