serializers.py

from rest_framework_jwt.serializers import JSONWebTokenSerializer
from rest_framework.serializers import ValidationError
from rest_framework.authtoken.serializers import AuthTokenSerializer

from .models import LoginLockout


class LockableTokenSerializer(object):
    """
    In addition to the normal token auth validation,
    validate that the user is not currently locked out from failed logins.
    """
    def validate(self, attrs):
        attrs = super(LockableTokenSerializer, self).validate(attrs)
        if LoginLockout.objects.user_is_locked_out(attrs['user']):
            raise ValidationError('User account is temporarily locked.')
        return attrs


class LockableJSONWebTokenSerializer(LockableTokenSerializer,
                                     JSONWebTokenSerializer):
    pass


class LockableAuthTokenSerializer(LockableTokenSerializer,
                                  AuthTokenSerializer):
    pass