Flask-Limiter 不会在 Flask-Restful API 的method_decorators上调用

Question

我试着跟随一些问题，但没有运气:

• flask - limitter不支持基于Flask-Restful api的应用
• 使用烧瓶限制器限速端点

由于outside of context问题，我在util.py文件中创建了一个limiter函数，如下所示:

def limiter():
_limiter = Limiter(
app,
key_func=get_remote_address
)
return _limiter

在我的Flask-Restful API资源中，我在method-decorators中调用了这个限制器:

from ..utils import limiter
class UsersView(Resource, CustomAPIMixin):
method_decorators = [limiter().limit("1/minute", methods=["GET"])]
@jwt_authenticate()
def get(self):
user_id = get_jwt_identity()
return self.api_response(content={"user_id": user_id})

注意:我使用Flask version 2和Flask-Limiter 1.4

---

EDIT-1:

my_api
models
scripts
views
users.py
__init__.py  # contains create_app() to return Flask app
utils.py. # contains custom rate_limit() function

---

EDIT-2:以下是Flask-Limiter无法工作的完整工作代码:

• https://github.com/alirezastack/ratelimit_api

Answer 1

Flask-Limiter文档给出了一个适合这种情况的示例:https://flask-limiter.readthedocs.io/en/stable/#using-flask-pluggable-views

如果你设置decorators而不是method_decorators，它应该解决你的问题。

下面是一个完整的工作示例，演示了所提供代码的速率限制。

from flask import Flask
from flask_restful import Resource, Api
from flask_limiter import Limiter
from flask_limiter.util import get_remote_address

app = Flask(__name__)
api = Api(app)

def limiter():
_limiter = Limiter(
app,
key_func=get_remote_address
)
return _limiter

class CustomAPIMixin:
def api_response(self, content):
return content

def get_jwt_identity():
return 0

class UsersView(Resource, CustomAPIMixin):
decorators = [limiter().limit("1/minute", methods=["GET"])]
# @jwt_authenticate()
def get(self):
user_id = get_jwt_identity()
return self.api_response(content={"user_id": user_id})

api.add_resource(UsersView, '/')
if __name__ == '__main__':
app.run(debug=True)