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_identity.py | 37 ++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 app/dispatch/management/commands/setup_identity.py (limited to 'app/dispatch/management/commands/setup_identity.py') 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() -- cgit v1.2.3