drf重写authenticate方法实现多条件登录(源码分析)1. 思路

JWT拓展的登录视图中, 在接受到用户名和密码时, 调用的也是Django的认证系统中提供的authenticate()来检查用户名与密码是否正确.

我们可以通过修改Django系统的认证后端来支持登录账号既可以是用户名也可以是手机号

(资料图片)

修改重写思路是:

修改Django认证系统的认证后端需要继承django.contrib.auth.backends.ModelBackends, 并重写authenticate方法

我们需要重写什么位置?

因为我们的需求是多条件登录, 所以我们就应该着手于数据库校验操作, 寻找定位源码中数据库校验username是否存在的操作, 并使用Q查询添加多个条件验证

2. 源码分析

下面是以from rest_framework_jwt.views import obtain_jwt_token为入口进行的源码分析(序号为查找流程):

继承JsonWebTokenAPIView视图, 当接受到用户post请求时, 执行get_serializer()

调用get_serializer()方法找到子类中定义的serializer类

当调用is_valid()方法时, 会调用到validate()

这里调用了authenticate方法并传入了用户数据, 返回的对象user

authenticate方法中调用了_get_backneds()方法, 此方法中将配置文件AUTHENTICATION_BACKENDS读取, 这个是django默认的配置, 我们重写方法之后, 需要将我们的类也配置到这里django的默认配置文件路径 django>conf>global_settings.py>AUTHENTICATION_BACKENDS读取了配置文件之后调用到了ModelBackend中的authenticate方法authenticate方法获取username携带username进行查表操作, 返回user对象如果查到user,就去验证其密码的正确性所以我们就可以重写的ModelBackend中的authenticate方法(修改数据库操作)3. 代码实现

在utils/authenticate.py中：

from django.contrib.auth.backends import ModelBackend, UserModelfrom django.db.models import Qdef get_user_by_account(account):    """    根据帐号信息获取user模型实例对象    :param account: 账号信息，可以是用户名，也可以是手机号，甚至其他的可用于识别用户身份的字段信息    :return: User对象 或者 None    """    user = UserModel.objects.filter(Q(mobile=account) | Q(username=account) | Q(email=account)).first()    return userclass CustomAuthBackend(ModelBackend):    """    自定义用户认证类[实现多条件登录]    """    def authenticate(self, request, username=None, password=None, **kwargs):        """        多条件认证方法        :param request: 本次客户端的http请求对象        :param username:  本次客户端提交的用户信息，可以是user，也可以mobile或其他唯一字段        :param password: 本次客户端提交的用户密码        :param kwargs: 额外参数        :return:        """        if username is None:            username = kwargs.get(UserModel.USERNAME_FIELD)        if username is None or password is None:            return        # 根据用户名信息useranme获取账户信息        user = get_user_by_account(username)        if user and user.check_password(password) and self.user_can_authenticate(user):            return user

在配置文件settings/dev.py中告知Django使用我们自定义的认证后端，注意不是给drf添加设置。

# django自定义认证AUTHENTICATION_BACKENDS = ["luffycityapi.utils.authenticate.CustomAuthBackend", ]

关键词：