From a691cfd2dace4eb67215cb1da35dc14585165d05 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=BE=9D=E7=91=AA=E8=B2=93?= Date: Wed, 17 May 2023 19:54:13 +0800 Subject: [PATCH] Applied the or_next utility to the set local route of the test site. --- tests/test_site/locale.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_site/locale.py b/tests/test_site/locale.py index 51273dd..202a50e 100644 --- a/tests/test_site/locale.py +++ b/tests/test_site/locale.py @@ -19,10 +19,12 @@ """ from babel import Locale from flask import request, session, current_app, Blueprint, Response, \ - redirect, url_for, Flask + redirect, Flask from flask_babel import Babel from werkzeug.datastructures import LanguageAccept +from accounting.utils.next_uri import or_next + bp: Blueprint = Blueprint("locale", __name__, url_prefix="/") @@ -68,9 +70,7 @@ def set_locale() -> Response: all_linguas: dict[str, str] = get_all_linguas() if "locale" in request.form and request.form["locale"] in all_linguas: session["locale"] = request.form["locale"] - if "next" in request.form: - return redirect(request.form["next"]) - return redirect(url_for("home.home")) + return redirect(or_next("/")) def get_all_linguas() -> dict[str, str]: