Flask UnboundLocalError问题：user_loader引用未赋值变量

错误原因分析 #

• 全局变量的局限性：Flask是多线程/多进程的WSGI应用，全局变量仅存在于当前进程内存中。注册时写入的user_data只在单个进程有效，登录请求可能被其他进程处理，导致找不到数据。
• 局部变量冲突：如果user_loader函数内部有对user_data的赋值操作，Python会将其视为局部变量，若在赋值前引用就会触发UnboundLocalError。

正确解决思路 #

放弃使用全局变量存储用户数据，改用数据库持久化存储（如SQLite、MySQL，甚至用内存字典模拟），在user_loader中通过用户ID从数据库读取对应数据，而非依赖注册时的临时变量。

代码示例 #

from flask import Flask, request, redirect, url_for
from flask_login import LoginManager, UserMixin, login_user, login_required, current_user

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

# 初始化登录管理器
login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'login'

# 模拟数据库（实际项目替换为真实数据库操作）
users_database = {}

# 定义用户类，继承UserMixin以实现登录所需的默认方法
class User(UserMixin):
    def __init__(self, user_data):
        self.id = user_data['_id']
        self.first_name = user_data['first_name']
        # 添加其他需要的用户字段

# 注册路由：收集用户信息并保存到数据库
@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        # 从表单获取注册数据
        first_name = request.form.get('first_name')
        # 生成唯一用户ID（实际可使用uuid或数据库自增ID）
        user_id = f"user_{len(users_database)+1}"
        
        # 组装用户数据
        user_data = {
            '_id': user_id,
            'first_name': first_name
            # 补充其他注册字段，如邮箱、密码哈希等
        }
        
        # 保存到数据库
        users_database[user_id] = user_data
        
        # 创建用户对象并完成登录
        user = User(user_data)
        login_user(user)
        return redirect(url_for('home'))
    return '''
        <form method="POST">
            <input type="text" name="first_name" placeholder="名字" required>
            <button type="submit">注册</button>
        </form>
    '''

# 用户加载回调：根据用户ID从数据库读取数据并返回User对象
@login_manager.user_loader
def load_user(user_id):
    # 从数据库获取对应ID的用户数据
    user_data = users_database.get(user_id)
    if user_data:
        return User(user_data)
    # 若找不到用户返回None
    return None

# 登录路由：验证用户身份并登录
@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        user_id = request.form.get('user_id')
        user_data = users_database.get(user_id)
        if user_data:
            user = User(user_data)
            login_user(user)
            return redirect(url_for('home'))
        return "用户不存在，请检查ID"
    return '''
        <form method="POST">
            <input type="text" name="user_id" placeholder="用户ID" required>
            <button type="submit">登录</button>
        </form>
    '''

# 首页：登录后可访问
@app.route('/home')
@login_required
def home():
    return f"欢迎回来，{current_user.first_name}！"

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

关键说明 #

1. 持久化存储：用数据库（示例中用内存字典模拟）保存所有用户数据，确保跨进程/线程能访问到。
2. user_loader的正确用法：该回调的核心是通过用户ID加载用户对象，必须从持久化存储中读取数据，而非依赖临时变量。
3. User类的作用：继承UserMixin可自动实现Flask-Login所需的get_id()、is_authenticated等方法，无需手动编写。

内容的提问来源于stack exchange，提问作者chiefqueef