diff --git a/main.py b/main.py index 80a85af..c0a8bb7 100755 --- a/main.py +++ b/main.py @@ -109,7 +109,8 @@ def login_page() -> str | werkzeug.Response: @app.route("/logout") def logout() -> werkzeug.Response: """Handle user logout by clearing the authentication cookie.""" - response = flask.redirect(flask.url_for("login_page")) + after_login = flask.request.args.get("next") + response = flask.redirect(flask.url_for("login_page", next=after_login)) response.set_cookie("auth_token", "", expires=0) flask.flash("You have been successfully logged out.", "info") return response