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
|
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'])
|
