From c8b9fe78f1c19b709345117bb8451c99576d213c Mon Sep 17 00:00:00 2001 From: Aevann Date: Sat, 7 Oct 2023 21:25:22 +0300 Subject: [PATCH] fix deleting messages in chat --- files/routes/chat.py | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/files/routes/chat.py b/files/routes/chat.py index e1a58e4e5..c3d0a315c 100644 --- a/files/routes/chat.py +++ b/files/routes/chat.py @@ -45,17 +45,20 @@ online = { cache.set('loggedin_chat', len(online[f'{SITE_FULL}/chat']), timeout=0) +def set_g_referrer(): + if request.referrer: + g.referrer = request.referrer.split('?')[0] + if g.referrer not in ALLOWED_REFERRERS: + g.referrer = None + else: + g.referrer = None + def auth_required_socketio(f): def wrapper(*args, **kwargs): v = get_logged_in_user() if not v: return '', 401 if v.is_permabanned: return '', 403 - if request.referrer: - g.referrer = request.referrer.split('?')[0] - if g.referrer not in ALLOWED_REFERRERS: - g.referrer = None - else: - g.referrer = None + set_g_referrer() return make_response(f(*args, v=v, **kwargs)) wrapper.__name__ = f.__name__ return wrapper @@ -65,12 +68,7 @@ def is_not_banned_socketio(f): v = get_logged_in_user() if not v: return '', 401 if v.is_suspended: return '', 403 - if request.referrer: - g.referrer = request.referrer.split('?')[0] - if g.referrer not in ALLOWED_REFERRERS: - g.referrer = None - else: - g.referrer = None + set_g_referrer() return make_response(f(*args, v=v, **kwargs)) wrapper.__name__ = f.__name__ return wrapper @@ -275,6 +273,8 @@ def typing_indicator(data, v): @socketio.on('delete') @admin_level_required(PERMS['POST_COMMENT_MODERATION']) def delete(id, v): + set_g_referrer() + if not g.referrer: return '', 400