mirror of https://github.com/interlegis/sapl.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.8 KiB
68 lines
1.8 KiB
#!/usr/bin/env python
|
|
import os
|
|
import sys
|
|
|
|
import django
|
|
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "sapl.settings")
|
|
|
|
|
|
def get_enviroment_admin_password(username):
|
|
password = os.environ.get("ADMIN_PASSWORD")
|
|
if not password:
|
|
print(
|
|
"[SUPERUSER] Environment variable $ADMIN_PASSWORD"
|
|
" for user %s was not set. Leaving..." % username
|
|
)
|
|
sys.exit("MISSING_ADMIN_PASSWORD")
|
|
return password
|
|
|
|
|
|
def create_user_interlegis():
|
|
from django.contrib.auth.models import User
|
|
|
|
password = get_enviroment_admin_password("interlegis")
|
|
print("[SUPERUSER INTERLEGIS] Creating interlegis superuser...")
|
|
user, created = User.objects.get_or_create(username="interlegis")
|
|
if not created:
|
|
print(
|
|
"[SUPERUSER INTERLEGIS] User interlegis already exists."
|
|
" Updating password."
|
|
)
|
|
user.is_superuser = True
|
|
user.is_staff = True
|
|
user.set_password(password)
|
|
user.save()
|
|
print("[SUPERUSER INTERLEGIS] Done.")
|
|
|
|
|
|
def create_superuser():
|
|
from django.contrib.auth.models import User
|
|
|
|
username = "admin"
|
|
email = os.environ.get("ADMIN_EMAIL", "")
|
|
|
|
if User.objects.filter(username=username).exists():
|
|
print(
|
|
"[SUPERUSER] User %s already exists." " Exiting without change." % username
|
|
)
|
|
sys.exit("ADMIN_USER_EXISTS")
|
|
else:
|
|
password = get_enviroment_admin_password(username)
|
|
|
|
print("[SUPERUSER] Creating superuser...")
|
|
|
|
u = User.objects.create_superuser(
|
|
username=username, password=password, email=email
|
|
)
|
|
u.save()
|
|
|
|
print("[SUPERUSER] Done.")
|
|
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
django.setup()
|
|
create_user_interlegis() # must come before create_superuser
|
|
create_superuser()
|
|
|