fix 400: improve 4e57369e68

a couple of other places shouldn't redirect either
pull/2/head
justcool393 2022-11-18 14:32:00 -06:00
parent 4e57369e68
commit 6f1b9014b4
1 changed files with 2 additions and 1 deletions

View File

@ -19,9 +19,10 @@ from files.routes.wrappers import *
@app.get("/login")
@auth_desired
def login_get(v):
NO_REDIRECT_URLS = ("/login", "/logout", "/signup", "/forgot", "/reset", "/reset_2fa", "/request_2fa_disable")
redir = request.values.get("redirect", "/").strip().rstrip('?')
if redir:
if not is_site_url(redir) or redir == "/reset_2fa":
if not is_site_url(redir) or redir in NO_REDIRECT_URLS:
redir = "/"
if v: return redirect(redir)