1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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()
|
