From f63237a9a2d9be2c8dc5a9953678d272ae94cfa1 Mon Sep 17 00:00:00 2001 From: justcool393 Date: Sun, 13 Nov 2022 02:36:36 -0600 Subject: [PATCH] fix ratelimit user to work with apps --- files/helpers/wrappers.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/files/helpers/wrappers.py b/files/helpers/wrappers.py index 62daedcb0..305e9414a 100644 --- a/files/helpers/wrappers.py +++ b/files/helpers/wrappers.py @@ -146,5 +146,9 @@ def feature_required(x): return wrapper return wrapper_maker -def ratelimit_user(limit=DEFAULT_RATELIMIT_USER): - return limiter.limit(limit, key_func=lambda:f'{SITE}-{session.get("lo_user")}') +def ratelimit_user(limit:Union[str, Callable[[], str]]=DEFAULT_RATELIMIT_USER): + ''' + Ratelimits based on a user. This requires at least auth_required (or stronger) to be present, + otherwise logged out users will receive 500s + ''' + return limiter.limit(limit, key_func=lambda:f'{SITE}-{g.v.id}')