flask扩展系列之 - 访问速度限制

flask-limiter 是一个对客户端的访问速率进行限制的flask扩展.可以自定义一些访问的(速度)限制条件来把那些触发限制的请求拒之门外.一般常用来进行对爬虫的限制.
下面就常见的用法,举了一些例子.更多的例子,请自己查看原始文档.

from flask import Flask, request, render_template
from flask_limiter import Limiter from flask_limiter.util import get_remote_address app = Flask(__name__) """ 默认的限制器 key_func参数是判断函数,表示以何种条件判断算一次访问?这里使用的是get_remote_address,此函数返回的是客户端的访问地址. default_limits 是一个数组,用于依次提同判断条件.比如100/day是指一天100次访问限制. 常用的访问限制字符串格式如下: 10 per hour 10/hour 10/hour;100/day;2000 per year 100/day, 500/7days 注意默认的限制器对所有视图都有效,除非你自定义一个限制器用来覆盖默认限制器,或者使用limiter.exempt装饰器来取消限制 """ limiter = Limiter(app=app, key_func=get_remote_address, default_limits=["100/day, 2/minute, 1/second"]) @limiter.request_filter def filter_func(): """ 定义一个限制器的过滤器函数,如果此函数返回True, 则不会施加任何限制.一般用这个函数创建访问速度 限制的白名单,可以使用某些celeb集中处理需要 limiter.exempt的情况 """ path_url = request.path white_list = ['/exempt'] if path_url in white_list: return True else: return False @app.route('/') @limiter.exempt # 取消默认限制器 def hello_world(): return render_template("login.html") @app.route("/exempt") def exempt_func(): """此视图函数被过滤器忽略""" return "no limited" @app.route('/h1', methods=['post', 'get']) # 自定义限制器覆盖了默认限制器 @limiter.limit("100/day;10/hour;3/minute") def h1_func(): return "h1" def get_key(): """自定义的key_fun函数""" return request.host_url @app.route('/h2', methods=['post', 'get']) # 自定义限制器覆盖了默认限制器,同时自定义了key_func函数 @limiter.limit(limit_value="100/day;10/hour;3/minute", key_func=get_key) def h2_func(): return "h2" @app.route('/h3', methods=['post', 'get']) # 自定义限制器覆盖了默认限制器,参数说明如下: # 1. param limit_value: 访问限制阈值 # 1. param exempt_when: 例外条件 (限制器只对使用不带debug访问的请求生效) # 2. param error_message: 错误的返回信息(注意,暂不支持中文,如果使用中文,请自定义错误页) # 3. param methods: 对哪些方法启用限制器? @limiter.limit(limit_value="1/minute", exempt_when=lambda: request.args.get("debug") is not None or request.form.get("debug") is not None, error_message="request be rejected", methods=['get']) def h3_func(): return "h3" """共享限制次数的限制器""" shared_limiter = limiter.shared_limit(limit_value="7/minute", scope="aaa") @app.route("/s1") # 和s2_func视图共享shared_limiter的限制阈值 @shared_limiter def s1_func(): return "s1" @app.route("/s2") # 和s1_func视图共享shared_limiter的限制阈值 @shared_limiter def s2_func(): return "s2" if __name__ == '__main__': app.run(host="0.0.0.0", port=5011, debug=True) 

原始文档地址 http://flask-limiter.readthedocs.io/en/stable/

转载于:https://www.cnblogs.com/ExMan/p/10912663.html