使用flask_limiter限制接口访问速率的方法

flask_limiter

目录

1.简介

这里简单介绍了一些使用flask_limiter的方法。具体详细可参考官方文档

https://flask-limiter.readthedocs.io/en/stable/

2.安装

shell 复制代码
pip install flask_socketio

3.初始化

python 复制代码
from flask import Flask
from flask_limiter import Limiter
from flask_limiter.util import get_remote_address

app = Flask(__name__)
limiter = Limiter(
    get_remote_address,
    app=app,
    default_limits=["200 per day", "50 per hour"],  # 这里是所有接口的默认访问上限限制
)

4.装饰器用法

在想要限制访问速率的函数前面加上装饰器,里面写限制规则,例如1 per day、1/second

示例:

python 复制代码
@app.route("/slow")
@limiter.limit("1 per day")
def slow():
    return ":("


@app.route("/medium")
@limiter.limit("1/second", override_defaults=False)
def medium():
    return ":|"


@app.route("/fast")
def fast():
    return ":)"


@app.route("/ping")
@limiter.exempt
def ping():
    return "PONG"

例如.limit("100/day;10/hour;1/minute")可以使用limiter单个装饰器设置多条规则

也可以使用多个装饰器设置多条规则

可以使用.limit("100/day", my_key_func)的方式调用自己的处理函数

python 复制代码
@limiter.limit("100/day;10/hour;1/minute")
def my_route():
    pass


@limiter.limit("100/day")
@limiter.limit("10/hour")
@limiter.limit("1/minute")
def my_route():
    pass


def my_key_func():
    pass

@limiter.limit("100/day", my_key_func)
def my_route():
    pass

5.limit用法

如果不希望使用装饰器,可以使用try加在函数里,捕获异常返回429

python 复制代码
@app.route("/")
def route():
   try:
       with limiter.limit("10/second"):
           # something expensive
   except RateLimitExceeded: pass
相关推荐
码出极致1 分钟前
Redisson分布式缓存与数据一致性保障
后端
用户790349033714 分钟前
springboot集成redisson实现redis分布式锁
后端
陈随易8 分钟前
程序员的新玩具,MoonBit(月兔)编程语言科普
前端·后端·程序员
码出极致14 分钟前
Redisson秒杀系统中的分布式锁应用
后端
xiaok22 分钟前
@Param注解的作用
java·后端
Sperains30 分钟前
async/await和Synchronous的区别
后端
码出极致33 分钟前
Redisson可重入锁(RLock)的使用与原理
后端
vx_bscxy32236 分钟前
springboot排课系统 -计算机毕业设计源码-23791
spring boot·后端·课程设计
im_AMBER1 小时前
python实践思路(草拟计划+方法)
开发语言·python
南雨北斗1 小时前
linux上TP6安装PHPMailer
后端