mirror of
https://github.com/netbox-community/netbox-docker.git
synced 2026-01-28 21:51:51 +00:00
23 lines
928 B
Python
23 lines
928 B
Python
|
from os import environ
|
||
|
|
from users.models import Token, User
|
||
|
|
from users.choices import TokenVersionChoices
|
||
|
|
from django.conf import settings
|
||
|
|
|
||
|
|
su_name = environ.get("SUPERUSER_NAME")
|
||
|
|
su_email = environ.get("SUPERUSER_EMAIL")
|
||
|
|
su_password = environ.get("SUPERUSER_PASSWORD")
|
||
|
|
su_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)
|
||
|
|
msg = ""
|
||
|
|
if not settings.API_TOKEN_PEPPERS:
|
||
|
|
print("⚠️ No API token will be created as API_TOKEN_PEPPERS is not set")
|
||
|
|
msg = f"💡 Superuser Username: {su_name}, E-Mail: {su_email}"
|
||
|
|
else:
|
||
|
|
t = Token.objects.create(
|
||
|
|
user=u, token=su_api_token, version=TokenVersionChoices.V2
|
||
|
|
)
|
||
|
|
msg = f"💡 Superuser Username: {su_name}, E-Mail: {su_email}, API Token: {t} (use with '{t.get_auth_header_prefix()}<Your token>')"
|
||
|
|
print(msg)
|