You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.6 KiB
45 lines
1.6 KiB
from django.contrib.auth import get_user_model
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from rest_framework import serializers
|
|
from rest_framework.authtoken.views import ObtainAuthToken
|
|
from rest_framework.compat import authenticate
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class AuthTokenSerializer(serializers.Serializer):
|
|
email = serializers.CharField(label=_("Email"))
|
|
password = serializers.CharField(
|
|
label=_("Password"),
|
|
style={'input_type': 'password'},
|
|
trim_whitespace=False
|
|
)
|
|
|
|
def validate(self, attrs):
|
|
email = attrs.get('email')
|
|
password = attrs.get('password')
|
|
|
|
if email and password:
|
|
user = authenticate(request=self.context.get('request'),
|
|
email=email, password=password)
|
|
|
|
# The authenticate call simply returns None for is_active=False
|
|
# users. (Assuming the default ModelBackend authentication
|
|
# backend.)
|
|
if not user:
|
|
msg = _('Unable to log in with provided credentials.')
|
|
raise serializers.ValidationError(msg, code='authorization')
|
|
elif user.role != User.ADMIN_ROLE:
|
|
msg = _('Only admin have permission to login admin page.')
|
|
raise serializers.ValidationError(msg, code='authorization')
|
|
else:
|
|
msg = _('Must include "email" and "password".')
|
|
raise serializers.ValidationError(msg, code='authorization')
|
|
|
|
attrs['user'] = user
|
|
return attrs
|
|
|
|
|
|
class ObtainToken(ObtainAuthToken):
|
|
serializer_class = AuthTokenSerializer
|
|
|