Cutlery/LibreTranslate
1
0
Fork 0
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2025-05-31 20:25:29 -04:00
Code Issues Packages Projects Releases Wiki Activity

Limiter func key with api key

Browse Source
This commit is contained in:
Piero Toffanin 2025-03-28 12:28:08 -04:00
parent c8f154561e
commit 6d43dfc504
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libretranslate/app.py
View File

@ -256,9 +256,18 @@ def create_app(args):
return max(req_cost, int(math.ceil(getattr(request, 'duration', 0) / args.req_time_cost))) return max(req_cost, int(math.ceil(getattr(request, 'duration', 0) / args.req_time_cost)))
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
), ),