server/ammicosrv/authentication/serializers.py
author ymh <ymh.work@gmail.com>
Wed, 08 Jan 2020 17:49:53 +0100
changeset 205 147583c43f0d
parent 156 bf4ae7d9a517
permissions -rw-r--r--
add poetry on sync
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
156
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     1
from django.contrib.auth import authenticate, get_user_model
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     2
from django.utils.translation import ugettext_lazy as _
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     3
from rest_framework import exceptions, serializers
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     4
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     5
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     6
class AuthTokenSerializer(serializers.Serializer):
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     7
    email = serializers.CharField()
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     8
    password = serializers.CharField(style={'input_type': 'password'})
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
     9
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    10
    def validate(self, attrs):
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    11
        email = attrs.get('email')
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    12
        password = attrs.get('password')
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    13
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    14
        if email and password:
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    15
            user = authenticate(email=email, password=password)
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    16
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    17
            if user:
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    18
                if not user.is_active:
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    19
                    msg = _('User account is disabled.')
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    20
                    raise exceptions.ValidationError(msg)
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    21
            else:
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    22
                msg = _('Unable to log in with provided credentials.')
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    23
                raise exceptions.ValidationError(msg)
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    24
        else:
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    25
            msg = _('Must include "email" and "password".')
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    26
            raise exceptions.ValidationError(msg)
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    27
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    28
        attrs['user'] = user
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    29
        return attrs
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    30
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    31
class UserSerializer(serializers.ModelSerializer):
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    32
    
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    33
    class Meta:
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    34
        model = get_user_model()
bf4ae7d9a517 refactor idUser to id_user
rougeronj
parents:
diff changeset
    35
        fields = ('id', 'email', 'last_login', 'id_user')