Limiter func key with api key

This commit is contained in:
Piero Toffanin 2025-03-28 12:28:08 -04:00
parent c8f154561e
commit 6d43dfc504

View file

@ -257,8 +257,17 @@ def create_app(args):
else: else:
return req_cost return req_cost
def get_limits_key_func():
if args.api_keys:
def func():
ak = get_req_api_key()
return ak if ak else get_remote_address()
return func
else:
return get_remote_address
limiter = Limiter( limiter = Limiter(
key_func=get_remote_address, key_func=get_limits_key_func(),
default_limits=get_routes_limits( default_limits=get_routes_limits(
args, api_keys_db args, api_keys_db
), ),