Flask中的内部重定向

Question

简而言之:

通过仅使用Flask微框架(及其依赖项),我们可以执行从一个路由到另一个路由的内部重定向吗？

例如:

1. 用户提交注册表(两者username和password)@app.route('/register', methods=['POST'])
2. 如果注册成功,Flask会在内部执行HTTP POST以@app.route('/login', methods['POST'])传递username和password
3. 处理并登录用户

---

细节:

我正在使用Flask和Flask-JWT扩展构建REST API .更具体地说,我正在实施登录和注册.

登录工作正常,并返回带有令牌的JSON对象.

以下是我的(登录)身份验证处理程序(即/auth(POST请求) - 默认Flask-JWT身份验证URL规则):

@jwt.authentication_handler
def authenticate(username, password):
    user = User.query.filter_by(username=username).first()
    if user and user.verify_password(password):
        return user
    return None

成功登录返回:

{
  "token": "<jwt-token>"
}

以下是我的注册路线:

@app.route('/register', methods=['PUT'])
def register():
    username = request.form.get('username')
    password = request.form.get('password')
    if username is None or password is None:
        abort(400)  # missing parameters

    user = User.query.filter_by(username=username).first()
    if user:
        abort(400)  # user exists
    else:
        user = User(user=user)
        user.hash_password(password)
        db.session.add(user)
        db.session.commit()

        # How do we generate a token?
        # Perform an internal redirect to the login route?

    return jsonify({'token': <jwt-token>}), 201

Answer 1

您应该使用 Post-Redirect-Get 模式。

from flask import Flask, redirect, request, render_template
app = Flask("the_flask_module")

@app.route('/', methods=["GET", "POST"])
def post_redirect_get():
    if request.method == "GET":
        return render_template("post_redirect_get.html")
    else:
        # Use said data.
        return redirect("target", code=303)

@app.route("/target")
def target():
    return "I'm the redirected function"

app.run(host="0.0.0.0", port=5001)

如果你想将数据传递给目标函数（比如那个令牌），你可以使用会话对象来存储它

所以那会分解类似的东西

@app.route('/register', methods=['PUT'])
def register():
    username = request.form.get('username')
    password = request.form.get('password')
    if username is None or password is None:
        abort(400)  # missing parameters

    user = User.query.filter_by(username=username).first()
    if user:
        abort(400)  # user exists
    else:
        user = User(user=user)
        user.hash_password(password)
        db.session.add(user)
        db.session.commit()

        # How do we generate a token?
        redirect("login_success", code=307)

@app.route("login_success", methods=["GET", "POST"])
@jwt_required()
def login_success():
    return "Redirected Success!"

编辑： 我之前没有使用过 Flask-JWT 并且不知道发布要求。但是您可以通过传递重定向函数code=307来告诉 Flask 使用当前使用的方法（而不是获取请求）进行重定向。. 希望这能解决您的扩展问题。