Try/except Knox Logout

This commit is contained in:
Daniel Stanton 2017-01-25 12:24:26 +00:00
parent 0db8c0dd95
commit fab9a7baa7

View File

@ -135,13 +135,13 @@ class LogoutView(APIView):
return self.logout(request) return self.logout(request)
def logout(self, request): def logout(self, request):
if getattr(settings, 'REST_USE_KNOX', False): try:
request._auth.delete() if getattr(settings, 'REST_USE_KNOX', False):
else: request._auth.delete()
try: else:
request.user.auth_token.delete() request.user.auth_token.delete()
except (AttributeError, ObjectDoesNotExist): except (AttributeError, ObjectDoesNotExist):
pass pass
django_logout(request) django_logout(request)