zoukankan      html  css  js  c++  java
  • 针对用户做限制(频率)

    可以针对用户名,ip地址做限制,应该也可以准备请求方式做限制

    方法1:

      

    from rest_framework.throttling import SimpleRateThrottle
    #第一步写一个频率类,继承SimpleRateThorttle
    # 重写get_cache_key,返回self.get_ident(request)
    # 要配置一个scop=字符串
    class Throttle(SimpleRateThrottle):
        scope = 'test'
        def get_cache_key(self, request, view):
            return self.get_ident(request)
    # 在setting中配置:
    REST_FRAMEWORK={
        "DEFAULT_THROTTLE_RATES":{
                'test':'3/m'
                                  }
    }
    #这里的设置是每分钟3次访问,可以设置访问次数,但是时间限制只能设置s m h 单位都是1 

    访问测试:

    当第四次访问的时候就会出现限制用户访问:

    缺点:方法已经定死,只能修改访问次数,不能修改具体的时间,只能使用1s、60s,3600s

    方法二(取消方法一的限制):

     加载模块:

    from rest_framework.throttling import SimpleRateThrottle, BaseThrottle

    设置自定义频率类(可以单独创建一个文件)

    # 自定义频率类
    class MyThrottle(BaseThrottle):
        # 设置空的字典
        visit_record = {}
        def __init__(self):
            # 定义一个列表
            self.history = None
        # 重写allow_request方法,优先继承本地
      #自定义控制每分钟访问多少次,运行访问返回true,不允许访问返回false
      # (1)取出访问者ip{ip1:[第二次访问时间,第一次访问时间],ip2:[]}
      # (2)判断当前ip不在访问字典里,如果不在添加进去,并且直接返回True,表示第一次访问,在字典里,继续往下走
      # (3)循环判断当前ip的列表,有值,并且当前时间减去列表的最后一个时间大于60s,把这种数据pop掉,这样列表中只有60s以内的访问时间,
      # (4)判断,当列表小于3,说明一分钟以内访问不足三次,把当前时间插入到列表第一个位置,返回True,顺利通过
      # (5)当大于等于3,说明一分钟内访问超过三次,返回False验证失败
        def allow_request(self, request, view):
            import time
            # 获取ip地址
            ip = request.META.get('REMOTE_ADDR')
            # 获取时间
            ctime = time.time()
            # 判断ip地址是否在字典中
            if ip not in self.visit_record:
            self.visit_record[ip]=[ctime]
                # 存在返回true
                return True
    
            print(self.visit_record.get(ip))
    
            print(self.history)
    
            self.history = self.visit_record.get(ip)
            # 最新事件减去最早事件大于60秒
            while self.history and ctime - self.history[-1] > 60:
                # 清空字典
                self.history.pop()
                # 字典长度小于3
            if len(self.history) < 3:
                # 在字典中插入新的时间
                self.history.insert(0, ctime)
                return True
            else:
                return False
    
        # 等待
        def wait(self):
            import time
            ctime = time.time()
            # 当前时间减去最早时间
            return 60 - (ctime - self.history[-1])

    其中检测次数和最长时间都可以手动设置

    通过request.META获取到的数据值中的一些类型都可以做限制,比如端口号,ip地址,请求方式之类,限制方法同上

    举例限制请求方式GET:

    获取get请求方式是:

    通过前端不同的请求方式判断频率限制是否生效:

    通过get方式请求测试:

    再次请求还是拒绝我们发起post请求试试

    由此得出结果:可以针对别的条件进行频率限制

    源码分析:

    SimpleRateThrottle与BaseThrottle从源码中我们可以看出是SimpleRateThrottle继承父类BaseThrottle

    我们定义一个自定义频率类,继承类BaseThrottle重写其中的一些判断方法:

    重写allow_request方法

    从这里就可以看出我们自定义的allow_request大致相同

    针对wait的return返回值中显示英文可以在源码

    
    
  • 相关阅读:
    markdown
    显示数学公式
    iOS----时间日期处理
    OC中文件读取类(NSFileHandle)介绍和常用使用方法
    深刻理解----修饰变量----关键字
    iOS----轻松掌握AFN网络顶级框架
    iOS
    iOS--多线程之线程间通讯
    iOS--多线程之NSOperation
    iOS--多线程之GCD
  • 原文地址:https://www.cnblogs.com/yangzhaon/p/11134874.html
Copyright © 2011-2022 走看看