如何访问 firebase 令牌并用作装饰器？

Question

我希望在访问之前对我的一些 flask 路由进行身份验证，这意味着任何带有装饰器 @check_token 的路由在访问路由之前都必须具有有效的令牌。

我有一个 login_or_create 路由，用户可以在其中登录或创建一个帐户到 Firebase。

从登录逻辑生成一个令牌，我想将该令牌传递给主路由。 我觉得我的代码快到了，但我不确定如何将令牌保留到下一条路线中。

def check_token(f):
    @wraps(f)
    def wrap(*args,**kwargs):
        if not request.headers.get('authorization'):
            return {'message': 'No token provided'},400
        try:
            user = auth.verify_id_token(request.headers['authorization'])
            request.user = user
        except:
            return {'message':'Invalid token provided.'},400
        return f(*args, **kwargs)
    return wrap

@app.route("/", methods=['GET', 'POST'])
def login_or_create():
    if request.method == 'POST':
        #Handle login form
        if request.form['action'] == 'login':
            data = request.form
            email = data.get('user_email')
            password = data.get('user_password')
            try:
                signin_user = pb.auth().sign_in_with_email_and_password(email, password)
                token = signin_user['idToken']
                return redirect(url_for('home'))
            except:
                return {'message':'There was an error logging in'}, 400



@app.route("/home")
@check_token
def home():
    return render_template('home_page.html)'

Answer 1

我最终将令牌保存在 session 变量中。 我认为这有缺点，但到目前为止，这一直是我的解决方案。