DRF - 访问频率

在 app 目录下创建 utils 目录，并在该目录下创建 throttle.py

throttle.py：

from rest_framework.throttling import BaseThrottle
import time


# 访问记录
VISIT_RECORD = {}


class MyThrottle(BaseThrottle):
    """
    每 10 秒访问 3 次
    """

    def __init__(self):
        self.history = None  # 用于记录访问记录

    def allow_request(self, request, view):
        # 获取用户的 IP 地址
        # 封装后的 request 中如果有的就不需要到 request._request 中找
        remote_addr = request.get_ident(request)
        ctime = time.time()  # 获取当前时间

        # 如果 IP 还没记录，即第一次访问
        if remote_addr not in VISIT_RECORD:
            # 记录第一次访问的 IP 地址和访问时间
            VISIT_RECORD[remote_addr] = [ctime, ]
            return True

        # 获取访问记录
        history = VISIT_RECORD.get(remote_addr)
        self.history = history

        # 把当前时间减 10 并和列表中最早的时间进行比较
        while history and history[-1] < ctime - 10:
            history.pop()  # 删除最早的一条记录

        # 如果还没有 3 条记录
        if len(history) < 3:
            # 将最新的访问时间插入到列表的第一个位置
            history.insert(0, ctime)
            print(history)
            return True

        return False

    # 设置剩余访问时间的提示
    def wait(self):
        ctime = time.time()
        return 10 - (ctime - self.history[-1])

history 列表的记录图示：

越早的访问记录放在列表的越后面的位置，方便操作

settings.py 中可以配置全局访问频率类：

REST_FRAMEWORK = {
    # 全局使用的访问频率类
    "DEFAULT_THROTTLE_CLASSES": ["drf.utils.throttle.MyThrottle"],
}

 views.py：

from django.http import JsonResponse
from rest_framework.views import APIView
from drf.utils.throttle import MyThrottle


ORDER_DICT = {
    1: {
        "commodity": "Phone",
        "price": 3600,
        "date": "2021-01-03",
    },
    2: {
        "commodity": "Computer",
        "price": 6700,
        "date": "2021-01-05",
    },
}


class OrderView(APIView):
    """
    查看订单
    """

    # 设置访问频率类
    throttle_classes = [MyThrottle, ]

    def get(self, request, *args, **kwargs):
        response = {"code": 1000, "msg": None, "data": None}
        try:
            response["data"] = ORDER_DICT
        except Exception as e:
            pass
        return JsonResponse(response)

10 秒内前 3 次可正常访问，访问第 4 次：