Django中Python Social Auth管道如何向视图传递正确后端名称？

有两种可靠的方式可以自动传递当前认证后端的名称到自定义视图，替代手动填写的方式：

方式一：通过Session存储后端名称 #

在管道步骤中将后端名称存入用户Session，后续视图从Session中取出使用，这种方式更安全，避免参数被篡改：

管道步骤代码 #

def check_password_required(backend, request, *args, **kwargs):
    # 假设这里判断需要中断管道，跳转到密码输入视图
    if not request.user.has_usable_password():
        # 将当前后端名称存入session
        request.session['social_backend_name'] = backend.name
        # 重定向到自定义密码输入视图
        return redirect('get_user_password')

自定义视图代码 #

def get_user_password(request):
    if request.method == 'POST':
        # 从session中取出后端名称，并移除该键避免重复使用
        backend_name = request.session.pop('social_backend_name', None)
        if backend_name:
            # 使用动态获取的后端名称生成complete路由
            return redirect(reverse('social:complete', args=(backend_name,)))
        # 处理后端名称不存在的异常情况
        return HttpResponseBadRequest("Invalid authentication backend")
    
    # GET请求渲染密码输入页面
    return render(request, 'auth/enter_password.html')

方式二：通过URL查询参数传递后端名称 #

如果不需要严格的安全性，也可以直接将后端名称作为查询参数附加到重定向URL中：

管道步骤代码 #

def check_password_required(backend, request, *args, **kwargs):
    if not request.user.has_usable_password():
        # 生成带后端名称参数的重定向URL
        redirect_url = f"{reverse('get_user_password')}?backend={backend.name}"
        return redirect(redirect_url)

自定义视图代码 #

def get_user_password(request):
    if request.method == 'POST':
        backend_name = request.GET.get('backend')
        if backend_name:
            return redirect(reverse('social:complete', args=(backend_name,)))
        return HttpResponseBadRequest("Invalid authentication backend")
    
    return render(request, 'auth/enter_password.html')

注意事项 #

• 使用Session方式时，记得在取出后端名称后调用pop移除对应键，防止用户重复提交或Session残留数据引发问题。
• 使用查询参数方式时，建议在视图中对backend_name做合法性校验（比如检查是否是系统支持的后端名称），避免恶意参数注入。

内容的提问来源于stack exchange，提问作者Tiago Peres