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()