forked from rDrama/rDrama
don't do it for the POST either
parent
2c81db79ef
commit
fc5d78a4a1
|
@ -16,13 +16,15 @@ from files.helpers.useractions import badge_grant
|
||||||
from files.routes.routehelpers import check_for_alts
|
from files.routes.routehelpers import check_for_alts
|
||||||
from files.routes.wrappers import *
|
from files.routes.wrappers import *
|
||||||
|
|
||||||
|
|
||||||
|
NO_LOGIN_REDIRECT_URLS = ("/login", "/logout", "/signup", "/forgot", "/reset", "/reset_2fa", "/request_2fa_disable")
|
||||||
|
|
||||||
@app.get("/login")
|
@app.get("/login")
|
||||||
@auth_desired
|
@auth_desired
|
||||||
def login_get(v):
|
def login_get(v):
|
||||||
NO_REDIRECT_URLS = ("/login", "/logout", "/signup", "/forgot", "/reset", "/reset_2fa", "/request_2fa_disable")
|
|
||||||
redir = request.values.get("redirect", "/").strip().rstrip('?')
|
redir = request.values.get("redirect", "/").strip().rstrip('?')
|
||||||
if redir:
|
if redir:
|
||||||
if not is_site_url(redir) or redir in NO_REDIRECT_URLS:
|
if not is_site_url(redir) or redir in NO_LOGIN_REDIRECT_URLS:
|
||||||
redir = "/"
|
redir = "/"
|
||||||
if v: return redirect(redir)
|
if v: return redirect(redir)
|
||||||
|
|
||||||
|
@ -105,7 +107,7 @@ def login_post():
|
||||||
|
|
||||||
redir = request.values.get("redirect", "").strip().rstrip('?')
|
redir = request.values.get("redirect", "").strip().rstrip('?')
|
||||||
if redir:
|
if redir:
|
||||||
if is_site_url(redir) and redir != "/reset_2fa":
|
if is_site_url(redir) and redir in NO_LOGIN_REDIRECT_URLS:
|
||||||
return redirect(redir)
|
return redirect(redir)
|
||||||
return redirect('/')
|
return redirect('/')
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue