33 lines
978 B
Python
33 lines
978 B
Python
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()
|