+
−"""
+
−Helper function for creating superusers in the authentication system.
+
−
+
−If run from the command line, this module lets you create a superuser
+
−interactively.
+
−"""
+
−
+
−from django.core import validators
+
−from django.contrib.auth.models import User
+
−import getpass
+
−import os
+
−import sys
+
−import re
+
−
+
−RE_VALID_USERNAME = re.compile('\w+$')
+
−
+
−def createsuperuser(username=None, email=None, password=None):
+
−    """
+
−    Helper function for creating a superuser from the command line. All
+
−    arguments are optional and will be prompted-for if invalid or not given.
+
−    """
+
−    try:
+
−        import pwd
+
−    except ImportError:
+
−        default_username = ''
+
−    else:
+
−        # Determine the current system user's username, to use as a default.
+
−        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 = ''
+
−
+
−    try:
+
−        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
+
−        while 1:
+
−            if not email:
+
−                email = raw_input('E-mail address: ')
+
−            try:
+
−                validators.isValidEmail(email, None)
+
−            except validators.ValidationError:
+
−                sys.stderr.write("Error: That e-mail address is invalid.\n")
+
−                email = None
+
−            else:
+
−                break
+
−        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)
+
−    u = User.objects.create_user(username, email, password)
+
−    u.is_staff = True
+
−    u.is_active = True
+
−    u.is_superuser = True
+
−    u.save()
+
−    print "Superuser created successfully."
+
−
+
−if __name__ == "__main__":
+
−    createsuperuser()