serializers.py
首先要写一份对应user的序列化类
from django.contrib.auth import get_user_model
from rest_framework import serializers
# 请务必使用get_user_model来获取django的user表
User = get_user_model()
class AuthUserSerializer(serializers.ModelSerializer):
username = serializers.CharField(max_length=15)
phone = serializers.CharField(max_length=11, min_length=11, error_messages={
"errorMsg": 'check your phone length, that must be 11'
}, required=False)
password = serializers.CharField(max_length=128, required=True)
class Meta:
model = User
fields = ["username", "phone", "password"]
# fields = '__all__' # fileds设置为'__all__'时,返回表模型所有字段
view.py
from django.db.models import Q
from rest_framework.exceptions import AuthenticationFailed
from django.contrib.auth import get_user_model
from django.contrib.auth.backends import ModelBackend
# 同上
User = get_user_model()
# 重写验证方式需要继承ModelBackend类
class CustomBackend(ModelBackend):
# 重写验证方法 这里的示例是验证username或者phone
def authenticate(self, request, username=None, password=None, **kwargs):
user = User.objects.get(Q(username=username) | Q(phone=username))
if not user:
raise AuthenticationFailed('用户不存在')
if user.check_password(password):
return user
raise AuthenticationFailed('密码错误')
settings.py
AUTHENTICATION_BACKENDS = (
'User.views.CustomBackend',
)
最后不要忘了在settings中修改默认的验证方法所在位置哦。