133 lines
5.4 KiB
Python
133 lines
5.4 KiB
Python
"""
|
|
Management utility to create superusers.
|
|
"""
|
|
|
|
import getpass
|
|
import os
|
|
import re
|
|
import sys
|
|
from optparse import make_option
|
|
from django.contrib.auth.models import User
|
|
from django.core import exceptions
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.utils.translation import ugettext as _
|
|
|
|
RE_VALID_USERNAME = re.compile('\w+$')
|
|
EMAIL_RE = re.compile(
|
|
r"(^[-!#$%&'*+/=?^_`{}|~0-9A-Z]+(\.[-!#$%&'*+/=?^_`{}|~0-9A-Z]+)*" # dot-atom
|
|
r'|^"([\001-\010\013\014\016-\037!#-\[\]-\177]|\\[\001-\011\013\014\016-\177])*"' # quoted-string
|
|
r')@(?:[A-Z0-9-]+\.)+[A-Z]{2,6}$', re.IGNORECASE) # domain
|
|
|
|
def is_valid_email(value):
|
|
if not EMAIL_RE.search(value):
|
|
raise exceptions.ValidationError(_('Enter a valid e-mail address.'))
|
|
|
|
class Command(BaseCommand):
|
|
option_list = BaseCommand.option_list + (
|
|
make_option('--username', dest='username', default=None,
|
|
help='Specifies the username for the superuser.'),
|
|
make_option('--email', dest='email', default=None,
|
|
help='Specifies the email address for the superuser.'),
|
|
make_option('--noinput', action='store_false', dest='interactive', default=True,
|
|
help='Tells Django to NOT prompt the user for input of any kind. ' \
|
|
'You must use --username and --email with --noinput, and ' \
|
|
'superusers created with --noinput will not be able to log in ' \
|
|
'until they\'re given a valid password.'),
|
|
)
|
|
help = 'Used to create a superuser.'
|
|
|
|
def handle(self, *args, **options):
|
|
username = options.get('username', None)
|
|
email = options.get('email', None)
|
|
interactive = options.get('interactive')
|
|
|
|
# Do quick and dirty validation if --noinput
|
|
if not interactive:
|
|
if not username or not email:
|
|
raise CommandError("You must use --username and --email with --noinput.")
|
|
if not RE_VALID_USERNAME.match(username):
|
|
raise CommandError("Invalid username. Use only letters, digits, and underscores")
|
|
try:
|
|
is_valid_email(email)
|
|
except exceptions.ValidationError:
|
|
raise CommandError("Invalid email address.")
|
|
|
|
password = ''
|
|
|
|
# Try to determine the current system user's username to use as a default.
|
|
try:
|
|
import pwd
|
|
except ImportError:
|
|
default_username = ''
|
|
else:
|
|
default_username = pwd.getpwuid(os.getuid())[0].replace(' ', '').lower()
|
|
|
|
# Determine whether the default username is taken, so we don't display
|
|
# it as an option.
|
|
if default_username:
|
|
try:
|
|
User.objects.get(username=default_username)
|
|
except User.DoesNotExist:
|
|
pass
|
|
else:
|
|
default_username = ''
|
|
|
|
# Prompt for username/email/password. Enclose this whole thing in a
|
|
# try/except to trap for a keyboard interrupt and exit gracefully.
|
|
if interactive:
|
|
try:
|
|
|
|
# Get a username
|
|
while 1:
|
|
if not username:
|
|
input_msg = 'Username'
|
|
if default_username:
|
|
input_msg += ' (Leave blank to use %r)' % default_username
|
|
username = raw_input(input_msg + ': ')
|
|
if default_username and username == '':
|
|
username = default_username
|
|
if not RE_VALID_USERNAME.match(username):
|
|
sys.stderr.write("Error: That username is invalid. Use only letters, digits and underscores.\n")
|
|
username = None
|
|
continue
|
|
try:
|
|
User.objects.get(username=username)
|
|
except User.DoesNotExist:
|
|
break
|
|
else:
|
|
sys.stderr.write("Error: That username is already taken.\n")
|
|
username = None
|
|
|
|
# Get an email
|
|
while 1:
|
|
if not email:
|
|
email = raw_input('E-mail address: ')
|
|
try:
|
|
is_valid_email(email)
|
|
except exceptions.ValidationError:
|
|
sys.stderr.write("Error: That e-mail address is invalid.\n")
|
|
email = None
|
|
else:
|
|
break
|
|
|
|
# Get a password
|
|
while 1:
|
|
if not password:
|
|
password = getpass.getpass()
|
|
password2 = getpass.getpass('Password (again): ')
|
|
if password != password2:
|
|
sys.stderr.write("Error: Your passwords didn't match.\n")
|
|
password = None
|
|
continue
|
|
if password.strip() == '':
|
|
sys.stderr.write("Error: Blank passwords aren't allowed.\n")
|
|
password = None
|
|
continue
|
|
break
|
|
except KeyboardInterrupt:
|
|
sys.stderr.write("\nOperation cancelled.\n")
|
|
sys.exit(1)
|
|
|
|
User.objects.create_superuser(username, email, password)
|
|
print "Superuser created successfully."
|