装饰不会被调用其他功能

问题描述 投票:1回答:1

看来,装饰(@need_to_login)没有在我的代码工作,因为当我打开网页就直接转到主页,而不是登录页面。这是下面的简单代码。如果有任何明显的错误,请告诉我。我是相当新的这一点。谢谢

from flask import Flask, render_template, redirect, url_for, request, session, flash
from functools import wraps

app = Flask(__name__)
app.secret_key='nelliville'

def need_to_login(f):
    @wraps(f)
    def wrap(*args, **kwargs):
        if 'logged_in' in session:
            return f(*args, **kwargs)
        else:
            flash('You need to log in first')
            return redirect(url_for('login'))
    return wrap

@app.route('/login', methods = ['GET', 'POST'])
def login():
    error = None
    if request.method =='POST':
        if request.form['username'] != 'admin' and request.form['password'] != 'admin':
            return 'Invalid credentials'    
        else:
            session['logged_in'] = True
            return redirect(url_for('home'))
    return render_template('login.html', error=error)

@app.route('/')
@need_to_login
def home():
    return render_template('home.html')

@app.route('/logout')
@need_to_login
def logout():
    session.pop('logged_in', None)
    return redirect(url_for('login'))

if __name__ == "__main__":
    app.run(debug=True)
decorator
1个回答
0
投票

好了,所以我不能发表评论,因为我没有足够的声誉作为即时通讯新的堆栈溢出。尝试在@wraps(f)去除need_to_login(f)

© www.soinside.com 2019 - 2024. All rights reserved.