diff --git a/files/__main__.py b/files/__main__.py index 8c107991a..6ffa62b71 100644 --- a/files/__main__.py +++ b/files/__main__.py @@ -5,7 +5,7 @@ from os import environ, path import secrets from flask import * from flask_caching import Cache -from flask_limiter import Limiter +import flask_limiter from flask_compress import Compress @@ -102,12 +102,9 @@ app.config["RATELIMIT_DEFAULTS_EXEMPT_WHEN"]=lambda:False app.config["RATELIMIT_HEADERS_ENABLED"]=True -def limiter_key_func(): return request.remote_addr - - -limiter = Limiter( +limiter = flask_limiter.Limiter( app, - key_func=limiter_key_func, + key_func=flask_limiter.util.get_ipaddr(), default_limits=["100/minute"], headers_enabled=True, strategy="fixed-window"