微信小程序-drf登录认证组件

微信小程序-drf登录认证组件

- 通过全局的用户判断是否登录
- 登录用户则校验用户的token值

1. config - settings.js

- api.js
- settings.js

模块化：登录页面路由

module.exports = {
loginPage: "/pages/login/login"
}

2. utils - auth.js

- auth.js

获取应用实例

var settings = require('../config/settings.js')
var app = getApp()

function authentication() {
if (!app.globalData.userInfo) {
wx.navigateTo({
// settings.loginPage,登录路由
url: settings.loginPage
})
return false
}
return true
}

认证函数模块化

module.exports = {
authentication: authentication
}

3. 示例：用户评论

pages
- newsDetail
- newsDetail.js

// pages/newsDetail/newsDetail.js

var api = require("../../config/api.js")
var auth = require('../../utils/auth.js')
var app = getApp()

评论之前判断用户是否登录

onClickShowCommentModal: function(e) {
if (!auth.authentication()) {
return
}
var replyInfo = e.currentTarget.dataset;
this.setData({
isShowCommentModal: true,
reply: replyInfo,
});
},

4. 提交评论-- 用户已登录，发送请求携带token值

'''
# 请求头携带用户的token值，此处涉及到js的三元运算
header: {
Authorization: userInfo ? "token " + userInfo.token : ""
},
'''

onClickPostComment: function() {
// 发布评论，将评论数据发送到后台接口
var userInfo = app.globalData.userInfo;
wx.request({
url: api.Comment,
data: this.data.reply,
method: 'POST',
dataType: 'json',
header: {
Authorization: userInfo ? "token " + userInfo.token : ""
},
responseType: 'text',
success: (res) => {
...
}
})
}

5. 后端校验 token值是否一致

utils
- auth.py

'''
如果用户已登录，则在request.user和request.auth中赋值；未登录则做任何操作。
用户需要在请求头Authorization中传递token，格式如下：
Authorization: token 401f7ac837da42b97f613d789819ff93537bee6a

建议：配合和配置文件一起使用，未认证的用户request.user和request.auth的值为None

REST_FRAMEWORK = {
"UNAUTHENTICATED_USER":None,
"UNAUTHENTICATED_TOKEN":None
}
'''

通用认证

!/usr/bin/env python

-- coding:utf-8 --

from rest_framework.authentication import BaseAuthentication
from rest_framework.authentication import get_authorization_header
from apps.api import models
from rest_framework import exceptions

class GeneralAuthentication(BaseAuthentication):

"""
通用认证（所有页面都可以应用）
"""

keyword = "token"

def authenticate(self, request):

认证元组

auth_tuple = get_authorization_header(request).split()

1.如果没有传token，则通过本次认证，进行之后的认证

if not auth_tuple:
return None

2.如果传递token，格式不符，则通过本次认证，进行之后的认证

if len(auth_tuple) != 2:
return None

3.如果传递了token，但token的名称不符，则通过本次认证，进行之后的认证

if auth_tuple[0].lower() != self.keyword.lower().encode():
return None

4.对token进行认证，如果通过了则给request.user和request.auth赋值，否则返回None

try:
token = auth_tuple[1].decode()
user_object = models.UserInfo.objects.get(token=token)
return (user_object, token)
except Exception as e:
return None

用户认证

class UserAuthentication(BaseAuthentication):

"""
token 认证
"""

keyword = "token"

def authenticate(self, request):
auth_tuple = get_authorization_header(request).split()

'''
print(auth_tuple)
auth_tuple = [b'token', b'a33409078e8ff69c16e813442ac1ce5d']
'''

if not auth_tuple:
raise exceptions.AuthenticationFailed('认证失败')

if len(auth_tuple) != 2:
raise exceptions.AuthenticationFailed('认证失败')

if auth_tuple[0].lower() != self.keyword.lower().encode():
raise exceptions.AuthenticationFailed('认证失败')
try:
token = auth_tuple[1].decode()
user_object = models.UserInfo.objects.get(token=token)
return (user_object, token)
except Exception as e:
raise exceptions.AuthenticationFailed('认证失败')

视图函数添加认证类 --评论

from utils.auth import UserAuthentication, GeneralAuthentication

class CommentView(CreateAPIView, ListAPIView):

'''
POST请求提交评论：用户认证类
'''
serializer_class = CommentModelSerializer
queryset = models.CommentRecord.objects

filter_backends = [ChildCommentFilter, ]

def get_authenticators(self):
if self.request.method == 'POST':
return [UserAuthentication(), ]
return [GeneralAuthentication(), ]