针对用户做限制(频率)

可以针对用户名,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返回值中显示英文可以在源码

 


posted @ 2019-07-04 20:58  阳光与叶子  阅读(343)  评论(0编辑  收藏  举报