fix: rate limiter key function (#2254)

* fix: rate limiter key function * lint
dpgaspar · Aug 19, 2024 · 1a85de9 · 1a85de9
1 parent 2d527aa
commit 1a85de9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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=get_remote_address)
+        limiter = Limiter(
+            key_func=app.config.get("RATELIMIT_KEY_FUNC", get_remote_address)
+        )
         limiter.init_app(app)
         return limiter