File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -14,8 +14,9 @@ _netbox_storages_map:
1414 sftpstorage : []
1515
1616netbox_superuser_script : |
17- from django.contrib.auth.models import User
17+ from django.contrib.auth import get_user_model
1818 from base64 import b64decode
19+ User = get_user_model()
1920 password = b64decode("{{ netbox_superuser_password | b64encode }}").decode("UTF-8")
2021 query = User.objects.filter(username="{{ netbox_superuser_username }}")
2122 if not query.exists():
@@ -33,7 +34,8 @@ netbox_superuser_script: |
3334
3435netbox_superuser_token : |
3536 from users.models import Token
36- from django.contrib.auth.models import User
37+ from django.contrib.auth import get_user_model
38+ User = get_user_model()
3739 user = User.objects.get(username="{{ netbox_superuser_username }}")
3840 query = Token.objects.filter(user=user)
3941 if not query.exists():
You can’t perform that action at this time.
0 commit comments