可以针对用户名,ip地址做限制,应该也可以准备请求方式做限制
方法1:
from rest_framework.throttling import SimpleRateThrottle #第一步写一个频率类,继承SimpleRateThorttle # 重写get_cache_key,返回self.get_ident(request) # 要配置一个scop=字符串 class Throttle(SimpleRateThrottle): scope = 'test' def get_cache_key(self, request, view): return self.get_ident(request) # 在setting中配置: REST_FRAMEWORK={ "DEFAULT_THROTTLE_RATES":{ 'test':'3/m' } } #这里的设置是每分钟3次访问,可以设置访问次数,但是时间限制只能设置s m h 单位都是1
访问测试:
当第四次访问的时候就会出现限制用户访问:
缺点:方法已经定死,只能修改访问次数,不能修改具体的时间,只能使用1s、60s,3600s
方法二(取消方法一的限制):
加载模块:
from rest_framework.throttling import SimpleRateThrottle, BaseThrottle
设置自定义频率类(可以单独创建一个文件)
# 自定义频率类 class MyThrottle(BaseThrottle): # 设置空的字典 visit_record = {} def __init__(self): # 定义一个列表 self.history = None # 重写allow_request方法,优先继承本地
#自定义控制每分钟访问多少次,运行访问返回true,不允许访问返回false
# (1)取出访问者ip{ip1:[第二次访问时间,第一次访问时间],ip2:[]}
# (2)判断当前ip不在访问字典里,如果不在添加进去,并且直接返回True,表示第一次访问,在字典里,继续往下走
# (3)循环判断当前ip的列表,有值,并且当前时间减去列表的最后一个时间大于60s,把这种数据pop掉,这样列表中只有60s以内的访问时间,
# (4)判断,当列表小于3,说明一分钟以内访问不足三次,把当前时间插入到列表第一个位置,返回True,顺利通过
# (5)当大于等于3,说明一分钟内访问超过三次,返回False验证失败
def allow_request(self, request, view): import time # 获取ip地址 ip = request.META.get('REMOTE_ADDR') # 获取时间 ctime = time.time() # 判断ip地址是否在字典中 if ip not in self.visit_record: self.visit_record[ip]=[ctime] # 存在返回true return True print(self.visit_record.get(ip)) print(self.history) self.history = self.visit_record.get(ip) # 最新事件减去最早事件大于60秒 while self.history and ctime - self.history[-1] > 60: # 清空字典 self.history.pop() # 字典长度小于3 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])
其中检测次数和最长时间都可以手动设置
通过request.META获取到的数据值中的一些类型都可以做限制,比如端口号,ip地址,请求方式之类,限制方法同上
举例限制请求方式GET:
获取get请求方式是:
通过前端不同的请求方式判断频率限制是否生效:
通过get方式请求测试:
再次请求还是拒绝我们发起post请求试试
由此得出结果:可以针对别的条件进行频率限制
源码分析:
SimpleRateThrottle与BaseThrottle从源码中我们可以看出是SimpleRateThrottle继承父类BaseThrottle
我们定义一个自定义频率类,继承类BaseThrottle重写其中的一些判断方法:
重写allow_request方法
从这里就可以看出我们自定义的allow_request大致相同
针对wait的return返回值中显示英文可以在源码