From f89a16f9c04c93049894b0e1de2922815eb45200 Mon Sep 17 00:00:00 2001 From: Daniel Gaspar Date: Tue, 2 Jul 2024 11:14:12 +0100 Subject: [PATCH 1/2] fix: rate limiter key function --- flask_appbuilder/security/manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flask_appbuilder/security/manager.py b/flask_appbuilder/security/manager.py index 82423cfa4c..5ca268029c 100644 --- a/flask_appbuilder/security/manager.py +++ b/flask_appbuilder/security/manager.py @@ -304,7 +304,7 @@ def __init__(self, appbuilder): self.limiter = self.create_limiter(app) def create_limiter(self, app: Flask) -> Limiter: - limiter = Limiter(key_func=get_remote_address) + limiter = Limiter(key_func=app.config.get("RATELIMIT_KEY_FUNC", get_remote_address)) limiter.init_app(app) return limiter From 96dcc7cc70452cb8c3a75248fee6f1a95268f324 Mon Sep 17 00:00:00 2001 From: Daniel Gaspar Date: Tue, 2 Jul 2024 13:42:22 +0100 Subject: [PATCH 2/2] lint --- flask_appbuilder/security/manager.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flask_appbuilder/security/manager.py b/flask_appbuilder/security/manager.py index 5ca268029c..d046eec9f0 100644 --- a/flask_appbuilder/security/manager.py +++ b/flask_appbuilder/security/manager.py @@ -304,7 +304,9 @@ def __init__(self, appbuilder): self.limiter = self.create_limiter(app) def create_limiter(self, app: Flask) -> Limiter: - limiter = Limiter(key_func=app.config.get("RATELIMIT_KEY_FUNC", get_remote_address)) + limiter = Limiter( + key_func=app.config.get("RATELIMIT_KEY_FUNC", get_remote_address) + ) limiter.init_app(app) return limiter