Django-rest-framework 接口实现 限制:(Throttle)

限制:(Throttle)

• 主要用来限制 单独一个用户的 访问次数

自定义一个 限制类

创建一个Throttle.py（截流）文件

• 注意 一定要写两个方法

  • def allow_request(self, request, view): allow(允许)

    • 定义限制的逻辑

  • def wait(self): **wait (等待) **

  • 定义被限制后的等待时间

import time
# 访问记录
visit_record = {}

class MyThrottle(object):
    # 限制访问时间
    VISIT_TIME = 10
    VISIT_COUNT = 3
	# 定义方法 方法名和参数不能变
    def allow_request(self, request, view):
        # 获取登录主机的id
        id = request.META.get('REMOTE_ADDR')
        self.now = time.time()

        if id not in visit_record:
            visit_record[id] = []

        self.history = visit_record[id]
		# 限制访问时间
        while self.history and self.now - self.history[-1] > self.VISIT_TIME:
            self.history.pop()
		# 此时 history中只保存了最近10秒钟的访问记录
        if len(self.history) >= self.VISIT_COUNT:
            return False
        else:
            self.history.insert(0, self.now)
            return True

    def wait(self):
        return self.history[-1] + self.VISIT_TIME - self.now

使用rest_framework 中已经定义好的 限制 直接配置即可

from rest_framework.throttling import SimpleRateThrottle
# 继承自带的类  定义全局配置 重写  def get_cache_key(self, request, view): 方法
class MyThrottle(SimpleRateThrottle):

    scope = "xxx"
    def get_cache_key(self, request, view):
        return self.get_ident(request)

• 从 rest_framework.throttling 中导入配置

• 常用 SimpleRateThrottle -- 简单的利率 多少时间内多少次

配置：在settings.py 中配置

全局配置:

REST_FRAMEWORK = {
    # 自定义限制类   也可以直接继承  DRF 中 自带的限制
    # 'DEFAULT_THROTTLE_CLASSES' = ['auth_demo.throttle.MyThrottle'],

    # 使用内置限制类  的额外配置
    "DEFAULT_THROTTLE_RATES": {
        # key  与定义的 scope 对应 value: 5 表示次数 / m表示分钟  s秒  h小时  d天
        "xxx": "5/m",
    }
}

局部配置:

from auth_demo.throttle import MyThrottle
# 写在要配置的 视图中
throttle_classes = [MyThrottle,]