【1126 | Day66】drf之频率组件

频率组件

为了防止恶意访问，需要对同一个访问者进行访问频率控制，借此减少服务器压力。

1. 频率组件工作原理

drf频率控制是通过单位时间内访问的次数来实现。

在drf的内部有一个字典，当ip访问的时候，有两种情况：

• 如果ip1是已经访问的ip，则在记录字典中添加这次的访问时间：{'ip1':[时间1，时间2，最新时间]}

• 如果IP2是第一次访问，则该字典进行添加操作：{'ip2':[最新访问时间]}

2. 源码阅读

3. 自定义频率控制类

from rest_framework.throttling import BaseThrottle
import time
VISIT_RECORD = {}
class MyThrotting(BaseThrottle):
    def __init__(self):
        self.history=[]
    #方法实现
    def allow_request(self, request, view):
        """
         以IP限流  一分钟同一个IP只能访问五次
        """
        #1：获取发访问一个ip
        ip = request.META.get("REMOTE_ADDR")
        '''
         2：判断当前ip是否存在，如果存在，添加最新的时间，如果不存在添加新的ip
        '''
        now_time=time.time()
        if ip in VISIT_RECORD:
            VISIT_RECORD[ip].insert(0, now_time)
        else:
            VISIT_RECORD[ip]=[now_time]
        history=VISIT_RECORD[ip]
        '''
        3:最后访问的和最新访问的时间相差一秒，删除最早访问的记录。保证字典内的记录均在一分钟以内。
        '''
        while history and history[0] - history[-1] > 60:
            history.pop()
        self.history=history
        """
        4：判断一分钟内，该ip访问次数是不是5次 
        """
        if len(history) > 5:
            return False
        else:
            return True
 
    def wait(self):
        """
        需要等待多长时间，才能再次访问
        """
        time = 60 - (self.history[0] - self.history[-1])
        return time

4. 局部配置和全局配置

5. 框架提供的频率认证组件的使用

a：导入包

from rest_framework.throttling import SimpleRateThrottle

b：settings.py配置

其他一些符号代表的时间频率：
('s', 'sec', 'm', 'min', 'h', 'hour', 'd', 'day')

REST_FRAMEWORK = {
     
    "DEFAULT_THROTTLE_RATES":{
        'WD':'5/m',         #速率配置每分钟不能超过5次访问，WD是scope定义的值，
 
    }
}

c：类的编写

from rest_framework.throttling import BaseThrottle,SimpleRateThrottle
class MyVisitThrottle(SimpleRateThrottle):
    #与settings配置的一样
    scope = "WD"
 
    def get_cache_key(self, request, view):
        return self.get_ident(request)

d：局部使用

class authDemo(APIView):
    throttle_classes=[MyVisitThrottle]
    def get(self, request):
        return Response('测试代码')