plateforme-tfjm2/apps/member/management/commands/create_su.py

import os
from datetime import date
from getpass import getpass
from django.core.management import BaseCommand
from member.models import TFJMUser


class Command(BaseCommand):
    def handle(self, *args, **options):
        """
        Little script that generate a superuser.
        """
        email = input("Email: ")
        password = "1"
        confirm_password = "2"
        while password != confirm_password:
            password = getpass("Password: ")
            confirm_password = getpass("Confirm password: ")
            if password != confirm_password:
                self.stderr.write(self.style.ERROR("Passwords don't match."))

        user = TFJMUser.objects.create(
            email=email,
            password="",
            role="admin",
            year=os.getenv("TFJM_YEAR", date.today().year),
            is_active=True,
            is_staff=True,
            is_superuser=True,
        )
        user.set_password(password)
        user.save()