encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-07 22:04:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Extend AuthTokenSerializer to allow login with custom username fields

Browse Source
This commit is contained in:
xtrinch 2016-06-12 23:57:34 +02:00
parent bb22ab8ee7
commit 9ba892f8d3
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
rest_framework/authtoken/serializers.py
View File

@ -2,18 +2,23 @@ from django.contrib.auth import authenticate
from django.utils.translation import ugettext_lazy as _ from django.utils.translation import ugettext_lazy as _
from rest_framework import serializers from rest_framework import serializers
from rest_framework.fields import empty
from django.contrib.auth import get_user_model
class AuthTokenSerializer(serializers.Serializer): class AuthTokenSerializer(serializers.Serializer):
username = serializers.CharField(label=_("Username"))
password = serializers.CharField(label=_("Password"), style={'input_type': 'password'}) password = serializers.CharField(label=_("Password"), style={'input_type': 'password'})
def __init__(self, data=empty, **kwargs):
self.fields[get_user_model().USERNAME_FIELD] = serializers.CharField(label=_(get_user_model().USERNAME_FIELD))
super().__init__(self, data, **kwargs)
def validate(self, attrs): def validate(self, attrs):
username = attrs.get('username') username = attrs.get('%s' % get_user_model().USERNAME_FIELD)
password = attrs.get('password') password = attrs.get('password')
if username and password: if username and password:
user = authenticate(username=username, password=password) user = authenticate(**{'%s' % get_user_model().USERNAME_FIELD: username, 'password': password})
if user: if user:
if not user.is_active: if not user.is_active:
@ -23,7 +28,7 @@ class AuthTokenSerializer(serializers.Serializer):
msg = _('Unable to log in with provided credentials.') msg = _('Unable to log in with provided credentials.')
raise serializers.ValidationError(msg) raise serializers.ValidationError(msg)
else: else:
msg = _('Must include "username" and "password".') msg = _('Must include "%s" and "password".' % get_user_model().USERNAME_FIELD)
raise serializers.ValidationError(msg) raise serializers.ValidationError(msg)
attrs['user'] = user attrs['user'] = user