I am using Flask-Security to set up user authentication for my app but I am struggling with getting the email address or ID of the currently logged in user so I can query a table with that particular users details. I’ve just using the standard code.
Another question here suggested the following but it didn’t work:
JavaScript
x
2
1
my_user = current_user.get_id()
2
Standard code:
JavaScript
1
53
53
1
from flask import Flask, render_template
2
from flask_sqlalchemy import SQLAlchemy
3
from flask_security import Security, SQLAlchemyUserDatastore,
4
UserMixin, RoleMixin, login_required
5
6
# Create app
7
app = Flask(__name__)
8
app.config['DEBUG'] = True
9
app.config['SECRET_KEY'] = 'super-secret'
10
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'
11
12
# Create database connection object
13
db = SQLAlchemy(app)
14
15
# Define models
16
roles_users = db.Table('roles_users',
17
db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),
18
db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))
19
20
class Role(db.Model, RoleMixin):
21
id = db.Column(db.Integer(), primary_key=True)
22
name = db.Column(db.String(80), unique=True)
23
description = db.Column(db.String(255))
24
25
class User(db.Model, UserMixin):
26
id = db.Column(db.Integer, primary_key=True)
27
email = db.Column(db.String(255), unique=True)
28
password = db.Column(db.String(255))
29
active = db.Column(db.Boolean())
30
confirmed_at = db.Column(db.DateTime())
31
roles = db.relationship('Role', secondary=roles_users,
32
backref=db.backref('users', lazy='dynamic'))
33
34
# Setup Flask-Security
35
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
36
security = Security(app, user_datastore)
37
38
# Create a user to test with
39
@app.before_first_request
40
def create_user():
41
db.create_all()
42
user_datastore.create_user(email='matt@nobien.net', password='password')
43
db.session.commit()
44
45
# Views
46
@app.route('/')
47
@login_required
48
def home():
49
return render_template('index.html')
50
51
if __name__ == '__main__':
52
app.run()
53
Advertisement
Answer
You can use flask_login.current_user
object. Its class will be what you configured Flask-Security to use to handle your users management, e.g. User
for code you included.