如何获取Flask+SQLAlchemy中已登录用户的用户名

嘿，这个问题我太熟了！在Flask里实现登录后调用用户名，核心就是利用**会话（session）**来跟踪用户身份，再结合你的User模型把用户信息带到需要的地方。下面分两种方案给你讲，从手动实现到用现成扩展简化，按需选择～

方案一：手动用Flask Session实现 #

1. 登录时把用户ID存入Session #

首先，你得在登录验证通过后，把用户的唯一标识（比如id）存入Flask的session对象里——别直接存用户名，用ID更安全，后续还能灵活获取用户的其他信息。

先确保你的Flask app设置了secret_key（session依赖它加密）：

from flask import Flask, session, request, redirect, url_for
from your_model_file import User, db

app = Flask(__name__)
app.secret_key = "your_strong_secret_key_here"  # 一定要换个复杂的密钥！

然后写登录视图：

@app.route("/login", methods=["GET", "POST"])
def login():
    if request.method == "POST":
        username = request.form["username"]
        password = request.form["password"]
        
        # 从数据库查用户（这里假设你有密码验证逻辑，别明文存密码！）
        user = User.query.filter_by(username=username).first()
        if user and user.password == password:  # 实际用加密验证，比如werkzeug的check_password_hash
            # 把用户ID存入session
            session["user_id"] = user.id
            return redirect(url_for("index"))
        else:
            return "用户名或密码错误"
    return render_template("login.html")

2. 全局注入当前用户信息 #

为了让模板和视图都能方便获取当前登录用户，用@app.context_processor把用户信息注入全局上下文：

@app.context_processor
def inject_current_user():
    current_user = None
    if "user_id" in session:
        # 通过session里的ID查数据库拿到用户对象
        current_user = User.query.get(session["user_id"])
    return {"current_user": current_user}

这样在模板里写博客发布表单时，就能直接用current_user.username了：

<form action="/create_post" method="POST">
    <input type="text" name="title" placeholder="标题">
    <textarea name="content" placeholder="内容"></textarea>
    <!-- 自动填充作者为当前登录用户名 -->
    <input type="hidden" name="author" value="{{ current_user.username }}">
    <button type="submit">发布文章</button>
</form>

在视图里处理文章发布也很简单：

@app.route("/create_post", methods=["POST"])
def create_post():
    # 先判断用户是否登录
    if "user_id" not in session:
        return redirect(url_for("login"))
    
    current_user = User.query.get(session["user_id"])
    title = request.form["title"]
    content = request.form["content"]
    
    # 假设你有Post模型，这里把用户名存为作者
    new_post = Post(title=title, content=content, author=current_user.username)
    db.session.add(new_post)
    db.session.commit()
    
    return redirect(url_for("post_list"))

3. 注销时清理Session #

别忘了写注销逻辑，把session里的用户ID清除：

@app.route("/logout")
def logout():
    session.pop("user_id", None)
    return redirect(url_for("login"))

方案二：用Flask-Login扩展简化（更推荐） #

如果不想自己写这些重复逻辑，Flask-Login这个官方推荐的扩展能帮你省很多事，它封装了登录状态管理、登录保护、current_user全局对象等功能。

1. 安装并初始化Flask-Login #

pip install flask-login

然后修改你的User模型，继承UserMixin（它帮你实现了Flask-Login需要的几个方法）：

from flask_login import UserMixin
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class User(Base, UserMixin):
    __tablename__ = "users"
    id = Column(Integer, primary_key=True)
    username = Column(String(64))
    password_hash = Column(String(128))  # 建议用加密后的密码存储

初始化Flask-Login：

from flask_login import LoginManager

login_manager = LoginManager()
login_manager.init_app(app)

# 告诉Flask-Login怎么通过user_id获取用户对象
@login_manager.user_loader
def load_user(user_id):
    return User.query.get(int(user_id))

2. 登录/注销视图简化 #

from flask_login import login_user, logout_user, login_required, current_user
from werkzeug.security import check_password_hash

@app.route("/login", methods=["GET", "POST"])
def login():
    if request.method == "POST":
        username = request.form["username"]
        user = User.query.filter_by(username=username).first()
        # 用加密验证密码
        if user and check_password_hash(user.password_hash, request.form["password"]):
            login_user(user)  # 直接用login_user方法，自动处理session
            return redirect(url_for("index"))
        return "登录失败"
    return render_template("login.html")

@app.route("/logout")
@login_required  # 只有登录用户才能访问注销页面
def logout():
    logout_user()  # 自动清理session
    return redirect(url_for("login"))

3. 发布博客的视图更简洁 #

用@login_required装饰器保护视图，确保只有登录用户能发布文章，直接用current_user获取当前用户：

@app.route("/create_post", methods=["POST"])
@login_required
def create_post():
    title = request.form["title"]
    content = request.form["content"]
    new_post = Post(title=title, content=content, author=current_user.username)
    db.session.add(new_post)
    db.session.commit()
    return redirect(url_for("post_list"))

模板里同样可以直接用current_user.username，和手动实现的用法一样～

注意事项 #

• 密码安全：绝对不要明文存储密码！用werkzeug.security.generate_password_hash()生成加密密码，check_password_hash()验证。
• secret_key要保密：不要把它硬编码到代码里，可以用环境变量加载。
• Flask-Login的login_required：可以给需要登录才能访问的视图加上这个装饰器，未登录用户会被重定向到登录页（你可以通过login_manager.login_view = "login"设置登录页路由）。

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