flask_login_user_session_with_protected_routes.py

1import flask
2from flask_login import LoginManager, login_user, logout_user, login_required, UserMixin
3
4app = flask.Flask(__name__)
5app.config['SECRET_KEY'] = 'use-a-secure-key-here'
6
7login_manager = LoginManager()
8login_manager.init_app(app)
9
10# Mock user database
11class User(UserMixin):
12    def __init__(self, id):
13        self.id = id
14
15users = {'user_id': User('user_id')}
16
17@login_manager.user_loader
18def load_user(user_id):
19    return users.get(user_id)
20
21@app.route('/login', methods=['GET', 'POST'])
22def login():
23    # In a real app, you would verify credentials here
24    user = users.get('user_id')
25    login_user(user)
26    return 'Logged in'
27
28@app.route('/logout')
29@login_required
30def logout():
31    logout_user()
32    return 'Logged out'
33
34@app.route('/protected')
35@login_required
36def protected():
37    return 'Logged in as: ' + flask.current_user.id
38
39if __name__ == '__main__':
40    app.run(debug=True)