DRF之频率

2022/4/9 6:22:23

本文主要是介绍DRF之频率,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

频率

  概念:

    认证、权限都通过以后,限制某个接口的访问频率,一般根据ip或者用户限制

  使用步骤

    第一步:

      写一个类,继承SimpleRateThrottle

      重写类属性:scope,和get_cache_key方法

      get_cache_key返回什么,就以什么做限制,scope配置文件中要用

    第二步::在配置文件中配置

REST_FRAMEWORK = {   
    'DEFAULT_THROTTLE_RATES': {
        'minute_3': '3/m'  # minute_3是scope的字符串,一分钟访问3次   
    }
}

    局部使用--视图类中

  class BookView(GenericViewSet, ListModelMixin, RetrieveModelMixin):
    throttle_classes = [IPThrottle]

    全局使用--配置文件中

 REST_FRAMEWORK = {
     'DEFAULT_THROTTLE_CLASSES': ['app01.auth.IPThrottle',],  # 全局配置频率类      
 }

  代码

  auth.py

from rest_framework.throttling import BaseThrottle, SimpleRateThrottle


class IpThrottle(SimpleRateThrottle):
    scope = 'minute_3'  # 随便写
    def get_cache_key(self, request, view):

        # 返回什么,就以什么做限制
        # 限制ip地址--request.META--请求中的数据都放在META中,他是个字典
        return request.META.get('REMOTE_ADDR')  # 客户端ip

 



这篇关于DRF之频率的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程