使用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
相关推荐
格林威17 分钟前
Baumer工业相机堡盟工业相机如何通过DeepOCR模型识别判断数值和字符串的范围和相似度(C#)
开发语言·人工智能·python·数码相机·计算机视觉·c#·视觉检测
赵英英俊24 分钟前
Python Day8
python
DAWN_T1735 分钟前
Python编译器(Pycharm Jupyter)
python·机器学习·jupyter·pycharm
Draina40 分钟前
在pycharm中运行sagemath脚本的配置过程
ide·python·安全·pycharm·密码学
presenttttt1 小时前
用Python和OpenCV从零搭建一个完整的双目视觉系统(五)
开发语言·python·opencv·计算机视觉
paopaokaka_luck1 小时前
基于Spring Boot+Vue的DIY手工社预约管理系统(Echarts图形化、腾讯地图API)
java·spring boot·后端
aramae1 小时前
Python3 -- 第二章 基本数据类型
笔记·python
kk_stoper1 小时前
使用Ruby接入实时行情API教程
java·开发语言·javascript·数据结构·后端·python·ruby
我会冲击波1 小时前
告别flag与status:如何为你的布尔值(boolean)变量优雅命名?
java·后端