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/