From c4fb52a2173862a2d4efee9d7163a8cab8e5d3a9 Mon Sep 17 00:00:00 2001 From: lare Date: Thu, 24 Nov 2022 16:02:07 +0100 Subject: [PATCH] fix KexError if no ?return= is specified on /login --- web/backend/main.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/web/backend/main.py b/web/backend/main.py index 9abca5a..8320fa0 100644 --- a/web/backend/main.py +++ b/web/backend/main.py @@ -128,9 +128,15 @@ def kioubit_auth(): if success: session["user-data"] = msg session["login"] = msg['mnt'] - return redirect(session["return_url"]) + try: + return redirect(session["return_url"]) + except KeyError: + return redirect(f"{config['base-dir']}peerings") else: - return render_template("login.html", session=session,config=config,return_addr=session["return_url"], msg=msg) + try: + return render_template("login.html", session=session,config=config,return_addr=session["return_url"], msg=msg) + except KeyError: + return render_template("login.html", session=session,config=config,return_addr=f"{config['base-dir']}peerings", msg=msg) @app.route("/logout") def logout():