Question

我正在使用Flask-Security为我的应用设置用户身份验证，但我正在努力获取当前登录用户的电子邮件地址或ID，以便我可以查询具有该特定用户详细信息的表。我刚刚使用标准代码。

另一个问题here建议如下，但它不起作用：

my_user = current_user.get_id()

标准代码：

from flask import Flask, render_template
from flask_sqlalchemy import SQLAlchemy
from flask_security import Security, SQLAlchemyUserDatastore, \
    UserMixin, RoleMixin, login_required

# Create app
app = Flask(__name__)
app.config['DEBUG'] = True
app.config['SECRET_KEY'] = 'super-secret'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'

# Create database connection object
db = SQLAlchemy(app)

# Define models
roles_users = db.Table('roles_users',
        db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),
        db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))

class Role(db.Model, RoleMixin):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    description = db.Column(db.String(255))

class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(255), unique=True)
    password = db.Column(db.String(255))
    active = db.Column(db.Boolean())
    confirmed_at = db.Column(db.DateTime())
    roles = db.relationship('Role', secondary=roles_users,
                            backref=db.backref('users', lazy='dynamic'))

# Setup Flask-Security
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
security = Security(app, user_datastore)

# Create a user to test with
@app.before_first_request
def create_user():
    db.create_all()
    user_datastore.create_user(email='matt@nobien.net', password='password')
    db.session.commit()

# Views
@app.route('/')
@login_required
def home():
    return render_template('index.html')

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

Answer 1

迈克尔是正确的一半，问题是标准代码示例不导入会话，虽然烧瓶安全在后端设置了会话，但它在烧瓶应用程序中不可用。迈克尔的代码：

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        session['username'] = request.form['username']
        session['email'] = request.form['email']

虽然在标准应用程序中正确可能会因烧瓶安全性而破坏，或者至少是不必要的，因为烧瓶安全性完全控制/登录路线并安全地处理表格。烧瓶示例应用程序所需的唯一两项更改是：

将烧瓶导入行更改为：

from flask import Flask, render_template, session

这是从会话中获取用户ID的示例：

@app.route('/dashboard')
@login_required
def dashboard():
    user_id = session["user_id"]
    return name

希望这可以帮助别人，因为我花了一段时间才能解决这个问题。

Answer 2

您可以使用flask_login.current_user对象。此类将是您配置的Flask-Security用于处理用户管理的内容，例如User中包含的代码。

Answer 3

恕我直言，你可以从Flask实施sessions。

from flask import Flask, session, redirect, url_for, escape, request

app = Flask(__name__)

@app.route('/')
def index():
    if 'username' in session:
        print("Currents user's ID is %s" % session['id']
        return 'Logged in as %s' % escape(session['username'])
    return 'You are not logged in'

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        session['username'] = request.form['username']
        session['email'] = request.form['email']
        session['id'] = request.form['id']
        return redirect(url_for('index'))
    return '''
        <form method="post">
            <p><input type=text name=username>
            <p><input type=submit value=Login>
        </form>
    '''

@app.route('/logout')
def logout():
    # remove the username from the session if it's there
    session.pop('username', None)
    session.pop('email', None)
    session.pop('id', None)
    return redirect(url_for('index'))

# set the secret key.  keep this really secret:
app.secret_key = 'A0Zr98j/3yX R~XHH!jmN]LWX/,?RT'

请参阅：http://flask.pocoo.org/docs/0.12/quickstart/#sessions

Answer 4

最好使用会话。您可以将信息存储在会话中，然后在任何地方使用它。

在您的登录功能中，只需存储以下值：首先从flask导入会话。然后像这样使用。

session['username'] = login_form.username.data

然后在模板中使用{{ session['username'] }}。

如何在实现Python Flask-Security时获得当前用户？

4 个答案: