Skip to content

Commit 1a85de9

Browse files
authored
fix: rate limiter key function (#2254)
* fix: rate limiter key function * lint
1 parent 2d527aa commit 1a85de9

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

flask_appbuilder/security/manager.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,9 @@ def __init__(self, appbuilder):
304304
self.limiter = self.create_limiter(app)
305305

306306
def create_limiter(self, app: Flask) -> Limiter:
307-
limiter = Limiter(key_func=get_remote_address)
307+
limiter = Limiter(
308+
key_func=app.config.get("RATELIMIT_KEY_FUNC", get_remote_address)
309+
)
308310
limiter.init_app(app)
309311
return limiter
310312

0 commit comments

Comments
 (0)