Fix authentication() returning None value

This commit is contained in:
Pieter Lambrecht 2022-03-18 10:37:28 +01:00
parent 14fbebfc24
commit 7d2ea49c01
2 changed files with 20 additions and 15 deletions

View File

@ -12,7 +12,9 @@ class TokenAuthentication(authentication.TokenAuthentication):
model = Token
def authenticate(self, request):
token_user, token = super().authenticate(request)
authenticationresult = super().authenticate(request)
if authenticationresult:
token_user, token = authenticationresult
# Verify source IP is allowed
if token.allowed_ips:
@ -28,6 +30,8 @@ class TokenAuthentication(authentication.TokenAuthentication):
raise exceptions.AuthenticationFailed(f"Source IP {clientip} is not allowed to use this token.")
return token_user, token
else:
return None
def authenticate_credentials(self, key):
model = self.get_model()

View File

@ -62,8 +62,9 @@ class TokenAdmin(admin.ModelAdmin):
]
def list_allowed_ips(self, obj):
if obj.allowed_ips:
return obj.allowed_ips
list_allowed_ips.empty_value_display = 'Any'
return 'Any'
list_allowed_ips.short_description = "Allowed IPs"