首页 > 编程语言 >自定义频率类,频率功能源码剖析,分页功能,排序功能,过滤功能

自定义频率类,频率功能源码剖析,分页功能,排序功能,过滤功能

时间:2022-10-10 22:22:24浏览次数:64  
标签:功能 return 自定义 self request rate num 频率 history

1.自定义频率类

继承BaseThrottle必须要重写allow_request方法,不然会报错

class OurThrottle(SimpleRateThrottle):
    scope = 'ss'

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


class UserThrottling(BaseThrottle):
    dic = {}

    def __init__(self):
        self.history = None

    def allow_request(self, request, view):
        ip = request.META.get('REMOTE_ADDR')
        import time
        timer = time.time()
        print(timer)
        if ip not in self.dic:
            self.dic[ip] = [timer, ]
            return True
        self.history = self.dic.get(ip)
        print(self.history)
        while self.history and timer - self.history[-1] > 60:
            # while self.history and -timer + self.history[-1] < 60:

            self.history.pop()
        if len(self.history) < 3:
            self.history.insert(0, timer)
            # print(self.history)
            return True
        else:
            return False

2.频率功能源码剖析

# SimpleRateThrottle
    -源码里执行的频率类的allow_request,读SimpleRateThrottle的allow_request
    
class SimpleRateThrottle(BaseThrottle):
    cache = default_cache
    timer = time.time
    cache_format = 'throttle_%(scope)s_%(ident)s'
    scope = None
    THROTTLE_RATES = api_settings.DEFAULT_THROTTLE_RATES
    def __init__(self):  # 只要类实例化得到对象就会执行,一执行,self.rate就有值了,而且self.num_requests和self.duration
        if not getattr(self, 'rate', None): # 去频率类中反射rate属性或方法,发现没有,返回了None,这个if判断就符合,执行下面的代码
            self.rate = self.get_rate()  #返回了  '3/m'
        #  self.num_requests=3
        #  self.duration=60
        self.num_requests, self.duration = self.parse_rate(self.rate)

    def get_rate(self):
         return self.THROTTLE_RATES[self.scope] # 字典取值,配置文件中咱们配置的字典{'ss': '3/m',},根据ss取到了 '3/m'

    def parse_rate(self, rate):
        if rate is None:
            return (None, None)
        # rate:字符串'3/m'  根据 / 切分,切成了 ['3','m']
        # num=3,period=m
        num, period = rate.split('/')
        # num_requests=3  数字3
        num_requests = int(num)
        # period='m'  ---->period[0]--->'m'
        # {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]]
        # duration=60
        duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]]
        # 3     60
        return (num_requests, duration)

    def allow_request(self, request, view):
        if self.rate is None:
            return True
        # 咱们自己写的,返回什么就以什么做限制  咱们返回的是ip地址
        # self.key=当前访问者的ip地址
        self.key = self.get_cache_key(request, view)
        if self.key is None:
            return True
        # self.history 访问者的时间列表,从缓存中拿到,如果拿不到就是空列表,如果之前有 [时间2,时间1]
        self.history = self.cache.get(self.key, [])
        # 当前时间
        self.now = self.timer()
        while self.history and self.history[-1] <= self.now - self.duration:
            self.history.pop()
        if len(self.history) >= self.num_requests:
            return self.throttle_failure()
        return self.throttle_success()
    
    
  # 总结:以后要再写频率类,只需要继承SimpleRateThrottle,重写get_cache_key,配置类属性scope,配置文件中配置一下就可以了

3.分页功能

一共有三种分页功能

from rest_framework.pagination import PageNumberPagination,LimitOffsetPagination,CursorPagination

class UserPageNumberPagination(PageNumberPagination):
    page_size = 2  #每页条数
    page_query_param = 'page' #查询第几页参数
    page_size_query_param = 'size'  #每页有几条数据
    max_page_size = 5  #最大不超过五条

class UserLimitOffsetPagination(LimitOffsetPagination):
    default_limit = 3   
    limit_query_param = 'limit'  #这一页显示4条数据
    offset_query_param = 'offset'  #从第几条数据往后开始显示
    max_limit = 5


class UserCursorPagination(CursorPagination):
    cursor_query_param = 'cursor'  
    page_size = 2
    ordering = '-id'  #根据什么进行排序

4 排序功能和过滤功能

#OrderingFilter是排序功能,然后在ordering_fields中写要排序的字段,最后在路由中加入就行
#SearchFilter是锅炉功能,同上

from rest_framework.filters import OrderingFilter,SearchFilter
filter_backends=[SearchFilter,OrderingFilter]
search_fields=['name']
ordering_fields=['id']

 

标签:功能,return,自定义,self,request,rate,num,频率,history
From: https://www.cnblogs.com/shangxin-bai/p/16777643.html

相关文章