mirror of
https://github.com/netbox-community/netbox-docker.git
synced 2026-03-18 14:26:53 +00:00
56909d6ef4
On v2 tokens, {t} prints the key not the token. Update the message
to say "API Key" and restore the auth header prefix hint.
38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
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 Key: {t}, (use with '{t.get_auth_header_prefix()}<Your token>')")
|