Python Flask用户登录重定向

from flask import Flask, redirect, url_for, render_template
from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user

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

# 初始化LoginManager
login_manager = LoginManager()
login_manager.init_app(app)

# 模拟用户数据
class User(UserMixin):
    def __init__(self, id):
        self.id = id

# 用户加载回调函数
@login_manager.user_loader
def load_user(user_id):
    return User(user_id)

# 登录页面
@app.route('/login')
def login():
    return render_template('login.html')

# 登录验证
@app.route('/login', methods=['POST'])
def login_post():
    # 假设用户名和密码验证通过
    user = User(1)
    login_user(user)
    return redirect(url_for('home'))

# 需要登录才能访问的页面
@app.route('/home')
@login_required
def home():
    return "Welcome to the home page!"

# 登出
@app.route('/logout')
@login_required
def logout():
    logout_user()
    return redirect(url_for('login'))

if __name__ == '__main__':
    app.run()