from optparse import make_option from django.core.management.base import BaseCommand from django.db import transaction from orchestra.apps.accounts.models import Account class Command(BaseCommand): def __init__(self, *args, **kwargs): super(Command, self).__init__(*args, **kwargs) self.option_list = BaseCommand.option_list + ( make_option('--noinput', action='store_false', dest='interactive', default=True), make_option('--username', action='store', dest='username'), make_option('--password', action='store', dest='password'), make_option('--email', action='store', dest='email'), ) option_list = BaseCommand.option_list help = 'Used to create an initial account.' @transaction.atomic def handle(self, *args, **options): interactive = options.get('interactive') if not interactive: email = options.get('email') username = options.get('username') password = options.get('password') Account.objects.create_user(username, email=email, password=password)