45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
|
from flask import Blueprint, render_template, request, redirect, url_for, flash, session
|
||
|
|
from flask import flash,redirect,url_for
|
||
|
|
from functools import wraps
|
||
|
|
from flask import session
|
||
|
|
|
||
|
|
class LoginAuth:
|
||
|
|
def __init__(self):
|
||
|
|
self.bp = Blueprint("auth", __name__)
|
||
|
|
|
||
|
|
# LOGIN ROUTE
|
||
|
|
@self.bp.route('/login', methods=['GET', 'POST'])
|
||
|
|
def login():
|
||
|
|
if request.method == 'POST':
|
||
|
|
username = request.form.get("username")
|
||
|
|
password = request.form.get("password")
|
||
|
|
|
||
|
|
# Dummy validation — REPLACE with DB check later
|
||
|
|
if username == "admin" and password == "admin123":
|
||
|
|
session['user'] = username
|
||
|
|
flash("Login successful!", "success")
|
||
|
|
return redirect(url_for('welcome'))
|
||
|
|
else:
|
||
|
|
flash("Invalid username or password!", "danger")
|
||
|
|
|
||
|
|
return render_template("login.html")
|
||
|
|
|
||
|
|
# LOGOUT ROUTE
|
||
|
|
@self.bp.route('/logout')
|
||
|
|
def logout():
|
||
|
|
session.clear()
|
||
|
|
flash("Logged out successfully!", "success")
|
||
|
|
return redirect(url_for('auth.login'))
|
||
|
|
|
||
|
|
# ===================================================
|
||
|
|
# LOGIN REQUIRED DECORATOR INSIDE CLASS
|
||
|
|
# ===================================================
|
||
|
|
def login_required(self, f):
|
||
|
|
@wraps(f)
|
||
|
|
def wrapper(*args, **kwargs):
|
||
|
|
if "user" not in session:
|
||
|
|
flash("Please login first!", "danger")
|
||
|
|
return redirect(url_for("auth.login"))
|
||
|
|
return f(*args, **kwargs)
|
||
|
|
return wrapper
|