day85

频率校验

源码分析

声明：基于rest_framework的频率校验

　　1.首先我们进入到APIView下的dispatch，因为由此方法开始分发的

　　　　2.可以看到dispatch方法下有一个initial的方法，进入该方法

　　3.由此进入频率控制

　　4.进入check_throttles方法，我们发现他最终是调用了该方法，所以我们需要在自定义时，写上这个方法

　　5.同时我们进入上张图中的迭代对象，发现与认证组件和权限控制一样他也需要在视图类中写一个列表

　　6.最后我们想他若是频率控制住后需要返回信息来到BaseThrottle,发现可以重写wait返回一个数字。

 

代码实现

 自定义频率控制

from rest_framework.throttling import BaseThrottle, SimpleRateThrottle
import time

class MyThrottle(BaseThrottle):
    dic = {'ip': []}

    def __init__(self):
        now_time = 0
        history = []

    def allow_request(self, request, view):
        self.now_time = time.time()
        ip = request.META.get("REMOTE_ADDR")
        if ip not in self.dic:
            self.dic[ip] = [self.now_time, ]
            return True
        self.history = self.dic.get(ip)
        while self.history and self.now_time - self.history[-1] > 60:
            self.history.pop()

        if len(self.history) < 3:
            self.history.insert(0, self.now_time)
            return True
        return False

    def wait(self):
        return 60 - (self.now_time - self.history[-1])

 视图函数

from django.shortcuts import render, HttpResponse
from rest_framework.views import APIView
from app01 import the
from rest_framework import exceptions



class User(APIView):
    throttle_classes = [the.MyThrottle, ]

    def get(self, request):
        return HttpResponse('get')

    def post(self, request):
        return HttpResponse('post')

 

使用rest_framework自带的频率控制

settting中设置

REST_FRAMEWORK = {
    'DEFAULT_THROTTLE_RATES': {
        'ttt': '3/m'
    }
}

在试图类中写一个继承Throttled的类可以控制返回信息为中文

    def throttled(self, request, wait):
        class Mythrottled(exceptions.Throttled):
            default_detail = '访问次数过多'
            extra_detail_singular = '剩余 {wait} 秒后访问.'
            extra_detail_plural = '剩余 {wait} 秒后访问.'

        raise Mythrottled(wait)

在py文件中设置相关值

class MyThrottle(SimpleRateThrottle):
    scope = 'ttt'

    def get_cache_key(self, request, view):
        # self.get_ident( request)
        return request.META.get('REMOTE_ADDR')

局部控制

局部使用:
	-在视图类中写
	    throttle_classes = [MyThrottle,]

全局控制

全局使用:
	-在setting中写
	    'DEFAULT_THROTTLE_CLASSES':['app01.MyAuth.MyThrottle',],
    -局部禁用:
	-在视图类中写
	    throttle_classes = []