码迷,mamicode.com
首页 > 其他好文 > 详细

Django rest-framework框架-访问频率控制

时间:2019-08-13 11:51:16      阅读:116      评论:0      收藏:0      [点我收藏+]

标签:response   ddr   _id   bsp   and   record   pytho   用户   cti   

第一版:

from rest_framework
class VisitThrottle(object): def __init__(self): self.history = None def allow_request(self,request,view): #获取用户IP #remote_addr = request.META.get(‘REMOTE_ADDR‘)
      remote_addr = self.get_ident(request)
ctime = time.time() if remote_addr not in VISIT_RECORD: VISIT_RECORD[remote_addr] = [ctime,] return True history = VISIT_RECORD.get(remote_addr) self.history = history while history and history[-1] < ctime - 10: #限制10秒内访问3次 history.pop() if length(history) < 3: history.insert(0,ctime) return True #return False 表示访问频率过高 限制访问 def wait(self): #等待多少秒 才能继续访问 ctime = time.time() return 60 - (ctime - self.history[-1]) class OrderView(APIView): """ 订单业务 """ authentication_classes = [] permission_classes = [] throttle_classes = [VisitThrottle,] def get(self, request, *args, **kwargs): ret = {‘code‘: 1000, ‘msg‘: None, ‘data‘: None} try: ret[‘data‘] = ORDER_DICT except Exception as e: pass return JsonResponse(ret)

  

全局配置:

"DEFAULT_THROTTLE_CLASSES": ["api.utils.throttle.UserThrottle"],
"DEFAULT_THROTTLE_RATES": {
                  "keyword": ‘3/m‘,     #限制1分钟内访问3次
                  "keyword2": ‘10/m‘,  #限制1分钟内访问10次
},    

  

第二版:

from rest_framework.throttle import SimpleRateThrottle

class VisitThrottle(SimpleRateThrottle):
        scope = "keyword"
    
        def get_cache_key(self, request, view):
              return self.get_ident(request) 

class UserThrottle(SimpleRateThrottle):
        scope = "keyword2"
    
        def get_cache_key(self, request, view):
              return request.user.username 
#scope 传递参数给 settings里面的做限制参数使用
"DEFAULT_THROTTLE_RATES": {
                  "keyword": ‘3/m‘,     #限制1分钟内访问3次
                  "keyword2": ‘10/m‘,  #限制1分钟内访问10次
}, 

  

第一版:

    - 必须继承 BaseThrottle

    - 必须实现 allow_request 和wait 方法

第二版:

     - 必须继承 SimpleRateThrottle 

     - 必须实现 get_cache_key方法 传递参数 scope = “userkey”

     - 在settings中 写入限制参数  "userkey": ‘10/m‘, #限制1分钟内访问10次

Django rest-framework框架-访问频率控制

标签:response   ddr   _id   bsp   and   record   pytho   用户   cti   

原文地址:https://www.cnblogs.com/kuku0223/p/11344906.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!