from functools import wraps
from flask import redirect, url_for, request, flash
from flask_login import current_user
from models import Employee  # Adjust the import path as needed

def employee_required(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        if not isinstance(current_user._get_current_object(), Employee):
            flash('❌ Please log in as an employee.', 'danger')
            return redirect(url_for('employee.employee_login', next=request.url))
        return f(*args, **kwargs)
    return decorated_function