Django之频率组件

一、频率简介

为了控制用户对某个url的请求 的频率，比如 ，一分钟以内，只能访问三次

二、自定义频率类，自定义频率规则

自定义的逻辑

（1）取出访问者的ip

（2）判断当前ip不在访问字典里，添加进去，并且直接返回True，表示第一次访问，在字典里，继续往下走

（3）循坏判断当前ip的列表，有值，并且当前时间减去列表的最后一时间大于60秒，把这种数据pop掉 ，这样列表中只有 60s以内的访问时间；

（4）判断，当列表小于3，说明一分钟 以内访问次数不足3次，把当前时间插入到列表第一个位置，返回True，顺利通过；

（5）当大于等于3，说明一分钟内访问超过3次，返回 False验证失败

代码实现：

import time
自定义频率控制
class MyThrottle():
    visitor_dic = {}

    def __init__(self):
        self.history = None

    def allow_request(self, request, view):
        '''

        #（1）取出访问者ip
        # （2）判断当前ip不在访问字典里，添加进去，并且直接返回True,表示第一次访问，在字典里，继续往下走
        # （3）循环判断当前ip的列表，有值，并且当前时间减去列表的最后一个时间大于60s，把这种数据pop掉，这样列表中只有60s以内的访问时间，
        # （4）判断，当列表小于3，说明一分钟以内访问不足三次，把当前时间插入到列表第一个位置，返回True，顺利通过
        # （5）当大于等于3，说明一分钟内访问超过三次，返回False验证失败
}        '''

        # META:请求所有的东西的字典
        # 拿出ip地址
        ip = request.META.get('REMOTE_ADDR')
        # 当前时间
        ctime = time.time()
        # self先从自身找，再到类中找
        if ip not in self.visitor_dic:
            self.visitor_dic[ip] = [ctime, ]
            return True
        # 根据当前时间者ip，取出访问的时间列表
        history = self.visitor_dic[ip]
        # 记录一下当前访问的人
        self.history = history
        while history and ctime - history[-1] > 60:
            history.pop()
        if len(history) < 3:
            # 将当前时间放到第0个位置上
            history.insert(0, ctime)
            return True
        else:
            return False

    def wait(self):
        # 剩余时间
        ctime = time.time()
        return 60 - (ctime - self.history[-1])

View Code