from django.core.management.base import BaseCommand, CommandError
from django.contrib.auth import get_user_model
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 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'])