healthchecks/hc/accounts/management/commands/createsuperuser.py

import getpass

from django.core.management.base import BaseCommand
from hc.accounts.forms import AvailableEmailForm
from hc.accounts.views import _make_user


class Command(BaseCommand):
    help = """Create a super-user account."""

    def handle(self, *args, **options):
        email = None
        password = None

        while not email:
            raw = input("Email address:")
            form = AvailableEmailForm({"identity": raw})
            if not form.is_valid():
                self.stderr.write("Error: " + " ".join(form.errors["identity"]))
                continue

            email = form.cleaned_data["identity"]

        while not password:
            p1 = getpass.getpass()
            p2 = getpass.getpass("Password (again):")
            if p1.strip() == "":
                self.stderr.write("Error: Blank passwords aren't allowed.")
                continue
            if p1 != p2:
                self.stderr.write("Error: Your passwords didn't match.")
                continue

            password = p1

        user = _make_user(email)
        user.set_password(password)
        user.is_staff = True
        user.is_superuser = True
        user.save()

        return "Superuser created successfully."