netbox-docker/docker/super_user.py

from os import environ

from django.conf import settings
from users.choices import TokenVersionChoices
from users.models import Token, User


# Read secret from file
def _read_secret(secret_name: str, default: str | None = None) -> str | None:
    try:
        f = open("/run/secrets/" + secret_name, "r", encoding="utf-8")
    except EnvironmentError:
        return default
    else:
        with f:
            return f.readline().strip()


su_name = environ.get("SUPERUSER_NAME", "admin")
su_email = environ.get("SUPERUSER_EMAIL", "admin@example.com")
su_password = _read_secret("superuser_password", environ.get("SUPERUSER_PASSWORD", "admin"))
su_api_token = _read_secret(
    "superuser_api_token",
    environ.get("SUPERUSER_API_TOKEN"),
)

if not User.objects.filter(username=su_name):
    u = User.objects.create_superuser(su_name, su_email, su_password)
    if not su_api_token:
        print("⚠️ No API token will be created as SUPERUSER_API_TOKEN is not set")
        print(f"💡 Superuser Username: {su_name}, E-Mail: {su_email}")
    elif not settings.API_TOKEN_PEPPERS:
        print("⚠️ No API token will be created as API_TOKEN_PEPPERS is not set")
        print(f"💡 Superuser Username: {su_name}, E-Mail: {su_email}")
    else:
        t = Token.objects.create(user=u, token=su_api_token, version=TokenVersionChoices.V2)
        print(f"💡 Superuser Username: {su_name}, E-Mail: {su_email}, API Token: {t}")