rest-framework之频率控制组件

频率组件的使用

# 写一个频率类，继承SimpleRateThrottle
class MyThrottle(SimpleRateThrottle):   
    scope = 'setting中设置的名字' # 一定要记住配置一个scop=字符串 
    # 重写get_cache_key方法
    def get_cache_key(self,request,view):
        return self.get_ident(request) # 返回self.get_ident(request)

# 要在setting中进行配置
 REST_FRAMEWORK = {
    'DEFAULT_THROTTLE_RATES': {
        'lxx': '3/minute'
    }
}

局部使用

# 在视图类中配置
throttle_classes=[Throttle,]

全局使用

# 在setting中配置   
'DEFAULT_THROTTLE_CLASSES':['自己定义的频率类'],

局部禁用

# 在视图类中配置
throttle_classes=[]

自定义的频率类

# 自定义频率类
class MyThrottle(BaseThrottle):
    VISIT_RECORD = {}

    def __init__(self):
        self.history = None

    def allow_request(self, request, view):
        # 自定义控制每分钟访问多少次，运行访问返回true，不允许访问返回false
        # （1）取出访问者ip
        # print(request.META)
        # 取出访问者ip
        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
        # 是个当前访问者ip对应的时间列表 [第一次访问的时间，]
        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])