标签: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