User registration, login, and logout with Flask-Login. Password hashing with bcrypt. Protecting routes with @login_required.
pip install flask-login flask-bcryptfrom flask_login import LoginManager, UserMixin, login_user, logout_user, login_required, current_user
from flask_bcrypt import Bcrypt
bcrypt = Bcrypt(app)
login_manager = LoginManager(app)
login_manager.login_view = 'login' # redirect here if not logged in
# User model must inherit UserMixin
class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True, nullable=False)
password_hash = db.Column(db.String(128), nullable=False)
@login_manager.user_loader
def load_user(user_id):
return User.query.get(int(user_id))@app.route('/register', methods=['GET','POST'])
def register():
if request.method == 'POST':
hashed = bcrypt.generate_password_hash(request.form['password']).decode('utf-8')
user = User(username=request.form['username'], password_hash=hashed)
db.session.add(user)
db.session.commit()
return redirect(url_for('login'))
return render_template('register.html')
@app.route('/login', methods=['GET','POST'])
def login():
if request.method == 'POST':
user = User.query.filter_by(username=request.form['username']).first()
if user and bcrypt.check_password_hash(user.password_hash, request.form['password']):
login_user(user, remember=True)
return redirect(url_for('dashboard'))
flash('Invalid credentials', 'danger')
return render_template('login.html')
@app.route('/logout')
@login_required
def logout():
logout_user()
return redirect(url_for('index'))
@app.route('/dashboard')
@login_required
def dashboard():
return render_template('dashboard.html', user=current_user)UserMixin adds is_authenticated, is_active, get_id() to your User model.generate_password_hash(pw) → stored hash. check_password_hash(hash, pw) → True/False.@login_required redirects unauthenticated users to login_manager.login_view.current_user is available in all views and templates when Flask-Login is set up.