initadmin.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os

import django

django.setup()

# Must come after django.setup or trigger "Apps aren't loaded yet" exception
from django.contrib.auth.models import User  # noqa


username = os.environ.get('DJANGO_SU_USERNAME', 'root')
password = os.environ.get('DJANGO_SU_PASSWORD', 'root')
email = os.environ.get('DJANGO_SU_EMAIL', 'root@localhost')

try:
    # if user exists allow password to be updated
    user = User.objects.get(username=username)
    user.set_password(password)
    user.is_superuser = True
    user.is_staff = True
    user.save()
except User.DoesNotExist:
    # otherwise create from scratch
    User.objects.create_superuser(
        username, email, password
    )