首页 > 技术文章 > rest_framework之认证与权限 token不存数据库认证

huangxuanya 2019-07-06 20:25 原文

1. 认证 :

介绍:

UserInfo表包含name , pwd , user_type三个字段
UserToken表包含token与user(关联UserInfo表)

当用户登录成功将随机字符串写入token, 并且将其返回给前台

当用户访问需要登录的视图需要将token拼到url中 , 如
http://127.0.0.1:8000/books/1?token=fbc4092d-b872-4b23-b2f7-29177ece9208
在自定义认证类里取到url的token, 与数据库的对比

源码分析

APIView---dispatch---initialize_request(包装request)---apiview的get_authenticators---
---将对象列表返回给authenticators----执行apiview的initial方法
---执行apiview的perform_authentication方法-----(request.user)
---Request.user方法----执行request对象的_authenticate方法------
循环request对象的authenticators----执行循环出的每个对象的authenticate方法,
将自己和request这个对象传入----最后返回一个元组(认证通过)
或者抛出异常(认证失败)exceptions.APIException--
--如果认证通过self.user, self.auth = user_auth_tuple执行的结果解压赋值,给request对象添加二个属性。

使用实例:

1. 自定义一个认证类 
注意:这个类要放在单独的py文件中,(如果放在view中,全局配置无法使用)
class LoginAuth():
    # 函数名一定要叫authenticate,接收必须两个参数,第二个参数是request对象
    def authenticate(self, request):
        # 从request对象中取出token(也可以从其它地方取)
        token = request.query_params.get('token')
        # 去数据库过滤,查询
        ret = models.UserToken.objects.filter(token=token).first()
        if ret:
            # 能查到,说明认证通过 返回两个值
            # ret.user就是当前登录用户对象,一旦retrun了,后面的认证类都不执行了
            return ret.user,ret
        # 如果查不到,抛异常
        raise exceptions.APIException('您认证失败')
        
2. 视图类中使用
- 局部使用
    在视图类中加一行:
        authentication_classes = [LoginAuth, ]
- 全局使用
    在setting中配置
        REST_FRAMEWORK={
            'DEFAULT_AUTHENTICATION_CLASSES':['app01.MyAuth.LoginAuth',]
        }
        中括号里的就是能导到认证类的东西
    局部禁用
        在视图类中加一行:
            authentication_classes = []

2. 权限

用法介绍

在自定义权限类里判断用户类型, 如果是3(超级会员)则权限足够, 返回Ture
如果权限不够返回False

代码实例

1. 自定义一个权限类
class UserPermission():
    # message是出错显示的中文
    message='您没有权限查看'
    def has_permission(self, request, view):
        user_type = request.user.user_type
        if user_type == 3:
            return True
        else:
            return False

2. 使用
-局部使用
    -在视图类中加一行:
     permission_classes = [UserPermission, ]

-全局使用
    -在setting中配置
    REST_FRAMEWORK={
        'DEFAULT_PERMISSION_CLASSES':['app01.MyAuth.UserPermission',]
    }
    -局部禁用
    -在视图类中加一行:
        -permission_classes = []

choice

choice的用法:
    -拿出数字对应的中文:get_字段名_dispaly()

====================================

认证组件的源码分析

执行到APIVIEW里面的dispatch方法 , 里面有一句
request = self.initialize_request(request, *args, **kwargs)
这一句会生成一个request对象, 
    生成这个对象的参数有
    django的request对象, 
    authenticators=self.get_authenticators(),
            这里的结果是authentication_classes列表里面的类实例化出来的一个个对象

然后会执行到self.initial(request, *args, **kwargs)这里
点进去里面有一句self.perform_authentication(request), 这个就是认证的

--------------------------------------------------------------------
|       分界线  上面的self代表视图类的对象  下面是Request的对象        |
--------------------------------------------------------------------

点进去里面调用了request.user, 这是一个包装过的方法, 在rest_framework.request.Request中
再进去lou一眼,里面有self._authenticate()方法,再点进去
里面会循环第六行里面的对象列表, 调用每个对象的authenticate(self)方法, 

注意!!!这里其实传了两个参数,
    第一个是对象调用方法自动将自己传过去了,
    第二个参数是request对象
源码里面会有两个变量来接收他的返回值,所以authenticate里面我们要返回两个值

总结:

综上, 所以我们需要在需要认证的视图类里面加上叫做authentication_classes的列表, 列表里面放入我们自定义的认证类

自定义认证类的时候里面需要加上方法def authenticate(self, request): ,并且返回两个值

在源码中这个方法被try了, 捕获的异常是exceptions.APIException , 所以如果认证失败抛出APIException这个异常

认证的token不存到数据库

每个用户登录都会往数据库里写一条数据,然后每次访问需要登录的地址时都要去数据库检查一下token, 这对数据库的压力比较大

现提供一种方法不往数据库写token验证

用户登陆成功以后用自己的方式加密一下用户的id

我用的是

"吃葡萄不吐葡萄皮 id 不吃葡萄倒吐葡萄皮"用md5加密,后面再拼上"|id"

将这个字符串作为token返回给前台

然后访问其他的需要登录的页面时url里拼上这个就好了

例如,: id为1的用户登录了, 生成一个7f38e7a645fbd1b3c68fbc75ecd62d24|1字符串

id为2的用户登录了, 生成一个a07687f76d66557e59db18bd89910e3a|2字符串

访问其他页面时带上这个字符串,例如访问书本详情时的url为:

http://127.0.0.1:8000/books/1?token=a07687f76d66557e59db18bd89910e3a|2

token翻译成明文就是吃葡萄不吐葡萄皮2不吃葡萄倒吐葡萄皮|2

在认证里面通过 " | " 切分取到id , 再通过上面的加密方式生成token与url里面的进行对比

代码示例:

获取加密字符串的函数
import hashlib
def get_token(id):
    md5=hashlib.md5()
    md5.update('吃葡萄不吐葡萄皮'.encode('utf-8'))
    md5.update(str(id).encode('utf-8'))
    md5.update('不吃葡萄倒吐葡萄皮'.encode('utf-8'))
    return md5.hexdigest()

登录成功以后生成token:
token = methord.get_token(user.id)+'|'+str(user.id)

认证类
class LoginAuth:
    def authenticate(self, request):
        token = request.GET.get('token')#获取url的token
        id = token.split('|')[1]#拿到id
        re_token = methord.get_token(id) + '|' + str(id)#根据我的加密方式进行加密
        if token == re_token:#判断url的token是否正确
            user=models.UserInfo.objects.filter(pk=int(id)).first()
            return user,user
        else:
            raise APIException('还没登录')

 

推荐阅读