From bb69819278d3316feba5772326f225f6d8611e6e Mon Sep 17 00:00:00 2001 From: jar3b Date: Fri, 6 Jul 2018 19:07:40 +0300 Subject: [PATCH] Fix auth --- drf_uuid_auth/authentication.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/drf_uuid_auth/authentication.py b/drf_uuid_auth/authentication.py index 97e367d..28d4aa0 100644 --- a/drf_uuid_auth/authentication.py +++ b/drf_uuid_auth/authentication.py @@ -1,4 +1,5 @@ from django.contrib.auth import get_user_model +from django.core.exceptions import FieldError, ValidationError from rest_framework import exceptions from rest_framework.authentication import BaseAuthentication @@ -20,9 +21,13 @@ class UuidAuthentication(BaseAuthentication): User = get_user_model() try: - user = User.object.get(uuid=payload) + user = User.objects.get(uid=payload) except User.DoesNotExist: raise exceptions.AuthenticationFailed("Invalid user id") + except FieldError: + raise exceptions.AuthenticationFailed("Incorrect user model") + except ValidationError: + raise exceptions.AuthenticationFailed("Incorrect user id") if not user.is_active: raise exceptions.AuthenticationFailed("User isn't active")