49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
 
|
"""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'))
|