skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

sneed

Browse source
This commit is contained in:
Aevann1 2021-07-22 22:31:24 +02:00
parent 922aa24a7b
commit e3a7e93dbb
1 changed files with 16 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

33
drama/routes/login.py
View file

@ -184,12 +184,6 @@ def sign_up_get(v):
return render_template("sign_up_failed_ref.html", return render_template("sign_up_failed_ref.html",
i=random_image()) i=random_image())
# check tor
# if request.headers.get("CF-IPCountry")=="T1":
# return render_template("sign_up_tor.html",
# i=random_image(),
# ref_user=ref_user)
# Make a unique form key valid for one account creation # Make a unique form key valid for one account creation
now = int(time.time()) now = int(time.time())
token = token_hex(16) token = token_hex(16)
@ -303,15 +297,14 @@ def sign_up_post(v):
# Check for existing accounts # Check for existing accounts
email = request.form.get("email") email = request.form.get("email")
email = email.strip() email = email.strip()
if not email: if not email: email = None
email = None
#counteract gmail username+2 and extra period tricks - convert submitted email to actual inbox #counteract gmail username+2 and extra period tricks - convert submitted email to actual inbox
if email and email.endswith("@gmail.com"): if email and email.endswith("@gmail.com"):
gmail_username=email.split('@')[0] email=email.split('@')[0]
gmail_username=gmail_username.split('+')[0] email=email.split('+')[0]
gmail_username=gmail_username.replace('.','') email=email.replace('.','')
email=f"{gmail_username}@gmail.com" email=f"{email}@gmail.com"
existing_account = get_user(request.form.get("username"), graceful=True) existing_account = get_user(request.form.get("username"), graceful=True)
@ -418,7 +411,13 @@ def post_forgot():
username = request.form.get("username").lstrip('@') username = request.form.get("username").lstrip('@')
email = request.form.get("email",'').strip() email = request.form.get("email",'').strip()
email=email.replace("_","\_").replace(".","") email=email.replace("_","\_")
if email.endswith("@gmail.com"):
email=email.split('@')[0]
email=email.split('+')[0]
email=email.replace('.','')
email=f"{email}@gmail.com"
user = g.db.query(User).filter( user = g.db.query(User).filter(
User.username.ilike(username), User.username.ilike(username),
@ -541,10 +540,10 @@ def request_2fa_disable():
email=request.form.get("email") email=request.form.get("email")
if email and email.endswith("@gmail.com"): if email and email.endswith("@gmail.com"):
gmail_username=email.split('@')[0] email=email.split('@')[0]
gmail_username=gmail_username.split('+')[0] email=email.split('+')[0]
gmail_username=gmail_username.replace('.','') email=email.replace('.','')
email=f"{gmail_username}@gmail.com" email=f"{email}@gmail.com"
if email != user.email: if email != user.email:
return render_template("message.html", return render_template("message.html",