博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
RESTful-rest_framework认证组件、权限组件、频率组件-第五篇
阅读量:4315 次
发布时间:2019-06-06

本文共 7591 字,大约阅读时间需要 25 分钟。

认证组件、权限组件、频率组件总结: 

认证组件格式:

1 写一个认证类        from rest_framework.authentication import BaseAuthentication        class MyAuth(BaseAuthentication):            def authenticate(self,request):        #         request  是封装后的                token = request.query_params.get('token')                ret = models.UserToken.objects.filter(token=token).first()                if ret:        #             认证通过                    return                else:                    raise AuthenticationFailed('认证失败')            #可以不写了            def authenticate_header(self,ss):                pass    2 局部使用        authentication_classes=[MyAuth,MyAuth2]    3 全局使用        查找顺序:自定义的APIView里找---》项目settings里找---》内置默认的        REST_FRAMEWORK={            'DEFAULT_AUTHENTICATION_CLASSES':['utils.common.MyAuth',]        }

 权限组件格式:

1 写一个类        class MyPermission():            def has_permission(self,request,view):                token=request.query_params.get('token')                ret=models.UserToken.objects.filter(token=token).first()                if ret.user.type==2:                # 超级用户可以访问                    return True                else:                    return False    2 局部使用:        permission_classes=[MyPermission,]    3 全局使用:            REST_FRAMEWORK={            'DEFAULT_PERMISSION_CLASSES':['utils.common.MyPermission',]        }

 频率组件格式:

1 写一个类:        from rest_framework.throttling import SimpleRateThrottle        class VisitThrottle(SimpleRateThrottle):            scope = 'xxx'            def get_cache_key(self, request, view):                return self.get_ident(request)    2 在setting里配置:            'DEFAULT_THROTTLE_RATES':{                'xxx':'5/h',            }    3 局部使用        throttle_classes=[VisitThrottle,]    4 全局使用        REST_FRAMEWORK={            'DEFAULT_THROTTLE_CLASSES':['utils.common.MyPermission',]        }

实例简介:

只有认证通过的用户才能访问指定的url地址,比如:查询课程信息,需要登录之后才能查看,没有登录,就不能查看,这时候需要用到认证组件

结构目录:

1.mode层

class UserInfo(models.Model):    name=models.CharField(max_length=32)    pwd=models.CharField(max_length=32)    ss=((1,'超级用户'),(2,'普通用户'),(3,'二逼用户'))    type=models.IntegerField(choices=ss,null=True)class UserToken(models.Model):    user=models.OneToOneField(to='UserInfo')    token=models.CharField(max_length=64)

2.views层

from django.shortcuts import render,HttpResponse# Create your views here.import jsonfrom rest_framework.views import APIViewfrom app01 import modelsfrom utils.common import *from rest_framework.response import Response#登录类class Login(APIView):    def post(self,request,*args,**kwargs):        #实例化响应状态函数(添加登录成功后的状态信息)        response=MyResponse()        #判断用户名、密码是否正确        name=request.data.get('name')        pwd=request.data.get('pwd')        user=models.UserInfo.objects.filter(name=name,pwd=pwd).first()        #如果登录成功生成一个随机字符串        if user:            #生成一个随机字符串            token=get_token(name)            #token表里面的信息,如果不存在,会创建,如果存在会更新token值(因为进行了随机时间加盐),使用的是update_or_create            ret=models.UserToken.objects.update_or_create(user=user,defaults={
'token':token}) # ret=models.UserInfo.objects.update_or_create(id=1,defaults={'token':token}) 不是只能写user或者可以写id response.status=100 response.msg='登录成功' response.token=token print(response.get_dic()) else: response.msg="用户名密码错误" # response.data='ddd' 最后response.get_dic(),都可以把这些信息返回 #里面需要传入个字典 return Response(response.get_dic())#查看课程类class Course(APIView): #局部登录认证 authentication_classes = [MyAuth,] #局部权限认证 permission_classes = [Mypermission,] #局部频率认证 throttle_classes = [VisitThrottle,] def get(self,request): print(request.user) print(request.auth) return HttpResponse(json.dumps({
'name':'python'}))

3.url与settings

#urlfrom django.conf.urls import urlfrom django.contrib import adminfrom app01 import viewsurlpatterns = [    url(r'^admin/', admin.site.urls),    url(r'^course/', views.Course.as_view()),    url(r'^login/', views.Login.as_view()),]#settings#全局使用认证REST_FRAMEWORK={    #登录认证,全局的登录认证刚开始不要加,如果添加后初始登录也会要求认证,那时数据库还没有数据,会有问题(所以可以在认证类里面设置局部的认证并设置为空)    # 'DEFAULT_AUTHENTICATION_CLASSES':['utils.common.MyAuth',],    #频率    'DEFAULT_THROTTLE_RATES':{            # 'xxx':'5/m',            'xxx':'5/hdddddddddddddddddddddddddddddd',        }}  例如解决全局配置后login初始登录会验证的问题:
class Login(APIView):     #为了可以在全局里面统一设置,并不影响初始这里的登录 authentication_classes=[]     def post(self,request,*args,**kwargs):     ...     ...

4.utils-common配置(组件类)

登录状态信息:

#账户登录状态信息class MyResponse():    def __init__(self):        #1001表示失败        self.status = 1001        self.msg = None    def get_dic(self):        #返回是字典格式的属性信息        return self.__dict__

随机验证码:

#传入用户名生成一个随机验证码tokendef get_token(name):    md = hashlib.md5()    md.update(name.encode('utf-8'))    md.update(str(time.time()).encode('utf-8'))    return md.hexdigest()

认证组件类:

from rest_framework.authentication import BaseAuthenticationfrom app01 import modelsfrom rest_framework.exceptions import APIException,AuthenticationFailed#认证组件class MyAuth(BaseAuthentication):    def authenticate(self, request):        #拿到的是随机token的值        token = request.query_params.get('token')        ret = models.UserToken.objects.filter(token=token).first()        if ret:            #认证通过            return ret.user,ret        else:            #认证失败            raise AuthenticationFailed('认证失败')

权限组件类:

#权限组件from rest_framework.permissions import BasePermissionclass Mypermission(BasePermission):    message = '不是超级用户,查看不了'    def has_permission(self, request, view):        token = request.query_params.get('token')        ret = models.UserToken.objects.filter(token=token).first()        #拿到userinfo表格对应的type字段,输出models里面userinfo表格里面type数字对应的“超级用户”等字符串信息        print(ret.user.get_type_display())        if ret.user.type==1:            return True        #如果是Flase的情况就会,打印上面的message信息        else:            return False

频率组件类:

#频率组件from rest_framework.throttling import SimpleRateThrottleclass VisitThrottle(SimpleRateThrottle):    scope = 'xxx'    def get_cache_key(self, request, view):        return self.get_ident(request)

运行效果展示:

准备工作:

1.userinfo数据表先添加2条数据,做演示

情况1:未登录情况下发送get请求,会提示认证失败(因为没有带token信息

情况2:post请求登录login,用户信息写入数据库,同时生成将出入的name+time盐,写入UserToken表token字段

 情况3:get请求携带token字符串,从而验证权限认证、频率认证

 

 

自定义频率类:

自定义逻辑

#(1)取出访问者ip# (2)判断当前ip不在访问字典里,添加进去,并且直接返回True,表示第一次访问,在字典里,继续往下走# (3)循环判断当前ip的列表,有值,并且当前时间减去列表的最后一个时间大于60s,把这种数据pop掉,这样列表中只有60s以内的访问时间,# (4)判断,当列表小于3,说明一分钟以内访问不足三次,把当前时间插入到列表第一个位置,返回True,顺利通过# (5)当大于等于3,说明一分钟内访问超过三次,返回False验证失败

  

class MyThrottles():    VISIT_RECORD = {}    def __init__(self):        self.history=None    def allow_request(self,request, view):        #(1)取出访问者ip        # print(request.META)        ip=request.META.get('REMOTE_ADDR')        import time        ctime=time.time()        # (2)判断当前ip不在访问字典里,添加进去,并且直接返回True,表示第一次访问        if ip not in self.VISIT_RECORD:            self.VISIT_RECORD[ip]=[ctime,]            return True        self.history=self.VISIT_RECORD.get(ip)        # (3)循环判断当前ip的列表,有值,并且当前时间减去列表的最后一个时间大于60s,把这种数据pop掉,这样列表中只有60s以内的访问时间,        while self.history and ctime-self.history[-1]>60:            self.history.pop()        # (4)判断,当列表小于3,说明一分钟以内访问不足三次,把当前时间插入到列表第一个位置,返回True,顺利通过        # (5)当大于等于3,说明一分钟内访问超过三次,返回False验证失败        if len(self.history)<3:            self.history.insert(0,ctime)            return True        else:            return False    def wait(self):        import time        ctime=time.time()        return 60-(ctime-self.history[-1])

 

转载于:https://www.cnblogs.com/yangzhizong/p/9833245.html

你可能感兴趣的文章
iOS 9音频应用播放音频之播放控制暂停停止前进后退的设置
查看>>
Delphi消息小记
查看>>
HNOI2016
查看>>
BZOJ2648: SJY摆棋子&&2716: [Violet 3]天使玩偶
查看>>
JVM介绍
查看>>
结构体,联合体,内存分配
查看>>
JVM垃圾收集器介绍
查看>>
[No0000136]6个重要的.NET概念:栈,堆,值类型,引用类型,装箱,拆箱
查看>>
【转】MapReduce源码分析总结
查看>>
原生JDK网络编程- NIO之Reactor模式
查看>>
table01
查看>>
简洁侧边wordpress博客模板
查看>>
linux及安全第四周总结——20135227黄晓妍
查看>>
搞出来,PHP mysql JQuery 二级联动
查看>>
AviSynth入门与应用指南
查看>>
ubuntu14.04安装GoldenDict
查看>>
重装系统时启动失败,引导信息有错误,修复磁盘的主引导记录MBR方法
查看>>
字符数组 字符指针
查看>>
Jedis的使用
查看>>
文献笔记(一)
查看>>