"""Authentication routes""" from flask import Blueprint, render_template, request, redirect, url_for, flash from flask_login import login_user, logout_user, login_required from app.models.user import User bp = Blueprint('auth', __name__, url_prefix='/auth') @bp.route('/login', methods=['GET', 'POST']) def login(): """User login page GET: Display login form POST: Process login credentials """ if request.method == 'POST': username = request.form.get('username', '').strip() password = request.form.get('password', '') if not username or not password: flash('Please provide both username and password', 'error') return render_template('auth/login.html') user = User.get_by_username(username) if user and user.is_active and user.check_password(password): login_user(user) flash(f'Welcome back, {user.username}!', 'success') # Redirect based on role if user.role == 'administrator': return redirect(url_for('admin.dashboard')) elif user.role == 'product_owner': return redirect(url_for('dashboard.list')) else: flash('Invalid username or password', 'error') return render_template('auth/login.html') @bp.route('/logout') @login_required def logout(): """User logout""" logout_user() flash('You have been logged out', 'info') return redirect(url_for('submission.form'))