File tree 1 file changed +4
-2
lines changed
1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -14,8 +14,9 @@ _netbox_storages_map:
14
14
sftpstorage : []
15
15
16
16
netbox_superuser_script : |
17
- from django.contrib.auth.models import User
17
+ from django.contrib.auth import get_user_model
18
18
from base64 import b64decode
19
+ User = get_user_model()
19
20
password = b64decode("{{ netbox_superuser_password | b64encode }}").decode("UTF-8")
20
21
query = User.objects.filter(username="{{ netbox_superuser_username }}")
21
22
if not query.exists():
@@ -33,7 +34,8 @@ netbox_superuser_script: |
33
34
34
35
netbox_superuser_token : |
35
36
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()
37
39
user = User.objects.get(username="{{ netbox_superuser_username }}")
38
40
query = Token.objects.filter(user=user)
39
41
if not query.exists():
You can’t perform that action at this time.
0 commit comments