session.py

# session.py

class Session:
    """Manages user session data."""

    current_user = None
    user_role = None

    @classmethod
    def login(cls, user_data):
        """Store logged-in user data."""
        cls.current_user = user_data
        cls.user_role = user_data.get('Role')

    @classmethod
    def logout(cls):
        """Clear session data."""
        cls.current_user = None
        cls.user_role = None

    @classmethod
    def is_logged_in(cls):
        """Check if user is logged in."""
        return cls.current_user is not None

    @classmethod
    def is_admin(cls):
        """Check if current user is admin."""
        return cls.user_role == 'admin'

    @classmethod
    def is_customer(cls):
        """Check if current user is customer."""
        return cls.user_role == 'customer'

    @classmethod
    def get_user_id(cls):
        """Get current user's ID."""
        if cls.current_user:
            return cls.current_user.get('UserID')
        return None

    @classmethod
    def get_username(cls):
        """Get current user's username."""
        if cls.current_user:
            return cls.current_user.get('Username')
        return None

    @classmethod
    def get_full_name(cls):
        """Get current user's full name."""
        if cls.current_user:
            fname = cls.current_user.get('FName', '')
            lname = cls.current_user.get('LName', '')
            return f"{fname} {lname}".strip()
        return None