From 324d1f5cd04c1af2f2b0aa341ef2571d0f6dd23d Mon Sep 17 00:00:00 2001 From: Aevann Date: Wed, 27 Sep 2023 00:40:06 +0300 Subject: [PATCH] make referrals work on original usernames --- files/routes/login.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/files/routes/login.py b/files/routes/login.py index 190076553..f26f5930a 100644 --- a/files/routes/login.py +++ b/files/routes/login.py @@ -151,13 +151,9 @@ def sign_up_get(v): abort(403, "New account registration is currently closed. Please come back later!") if v: return redirect(SITE_FULL) - ref = request.values.get("ref") + ref = request.values.get("ref", "") - if ref: - ref = ref.replace('\\', '').replace('_', '\_').replace('%', '').strip() - ref_user = g.db.query(User).filter(User.username.ilike(ref)).one_or_none() - else: - ref_user = None + ref_user = get_user(ref) if ref_user and (ref_user.id in session.get("history", [])): return render_template("login/sign_up_failed_ref.html"), 403