Rest_Framework之频率组件部分

一、RestFramework之频率组件源码部分

频率组件的源码部分和权限组件流程一模一样的，这里就不多说了，直接上源码的主要逻辑部分：

def check_throttles(self, request):"""Check if request should be throttled.Raises an appropriate exception if the request is throttled."""for throttle in self.get_throttles():if not throttle.allow_request(request, self):self.throttled(request, throttle.wait())

明确表示我们写的频率类需要一个allow_request()方法：

频率类（完成一分钟同一个ip只能访问三次）：

import time
from rest_framework.throttling import BaseThrottleclass MyThrottle(BaseThrottle):visited_record = {}def __init__(self):self.history = Nonedef allow_request(self, request, my_cbv):# 这个my_cbv是源码中传的我们的视图类，这里我们也要传进去# print(self.get_ident(request))  # 可以获取本次请求的ipip = request.META.get("REMOTE_ADDR")if ip not in self.visited_record:self.visited_record[ip] = []current_time = time.time()history = self.visited_record[ip]self.history = historywhile history and current_time - history[-1] > 60:  # 把与当前访问时间相差大于60秒的时间都删掉
            history.pop()if len(history) > 2:  # 第三次访问，列表中只有2个值，也满足条件，大于2个值时不满足条件return Falsehistory.insert(0, current_time)return Truedef wait(self):"""用于返回还剩多少时间访问；本次访问时间：9:50:55[09:50:30, 09:50:20, 09:50:10]   剩余 60 - (9:50:55 - 09:50:10)秒才能访问:return:"""c_time = time.time()return 60 - (c_time - self.history[-1])  

视图类：

class BookView(ModelViewSet):authentication_classes = [UserAuth]      #认证类permission_classes = [UserPerm]        #权限类throttle_classes = [MyThrottle]　　　　　　#频率类queryset = Book.objects.all()serializer_class = BookSerializer

效果如下：

可以在全局settings配置

REST_FRAMEWORK = {'DEFAULT_PARSER_CLASSES': ('rest_framework.parsers.JSONParser','rest_framework.parsers.FormParser','rest_framework.parsers.MultiPartParser'),'DEFAULT_AUTHENTICATION_CLASSES': 'app01.utils.auth_class.UserAuth',),'DEFAULT_PERMISSION_CLASSES': ('app01.utils.permission_class.VipPermission',),'DEFAULT_THROTTLE_CLASSES': ('app01.utils.throttle_class.MyThrottle',),
}

 二、使用restframework组件中的提供的访问限制

实现方式和我们上面的方式基本相同；

基于限制ip的类：SimpleRateThrottle   

基于ip的访问限制：

频率类——局部：

from rest_framework.throttling import SimpleRateThrottleclass MyThrottle(SimpleRateThrottle):rate = '5/m'def get_cache_key(self, request, view):  # 这个方法也是必须要有return self.get_ident(request)

 

在视图类中指定频率类

class BookView(ModelViewSet):throttle_classes = [app_throttles.RateThrottle]queryset = Book.objects.all()serializer_class = BookSerializer

 

 duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}

频率类——全局：

from rest_framework.throttling import SimpleRateThrottleclass MyThrottle(SimpleRateThrottle):scope = "visit_rate"  # 这个值决定了在配置时使用哪个变量描述限制的频率,必须在settings里面配置def get_cache_key(self, request, view):  # 这个方法也是必须要有return self.get_ident(request)

这次只能在setttings中配置：

REST_FRAMEWORK = {'DEFAULT_THROTTLE_CLASSES': ('app01.utils.throttle_class.MyThrottle',),"DEFAULT_THROTTLE_RATES": {"visit_rate": "10/m",   # 这个参数就是频率类中定义的那个参数scope， 其中第一个数字10表示10次，后面的m表示一分钟，还有s，一秒， h, 一小时， d, 一天
    }
}

 duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}