rDrama
/
rDrama
9
10
Fork 7
Code Issues Pull Requests Packages Releases Wiki Activity

make referrals work on original usernames

pull/210/head
Aevann 2023-09-27 00:40:06 +03:00
parent c9aa92cc04
commit 324d1f5cd0
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
files/routes/login.py
View File

@ -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