state_manager.py

class StateManager:
    
    def __init__(self):
        self.states = {} # Diccionario de estados
        self.state = None # Estado actual none por defecto
        self.current_state = None
        self.selected_level = None
        self.selected_character = 0
        self.difficulty = "Beginner"
        
    def add_state(self, state_name, state):
        self.states[state_name] = state # Añade un estado al diccionario de estados
        
    def change_state(self, new_state):
        self.current_state = new_state    
        
    def set_state(self, state_name):
        state_class = self.states.get(state_name)
        if state_class:
            self.current_state = state_class(self)
            if state_name.startswith("level") and state_name.replace("level", "").isdigit():
                self.current_level = int(state_name.replace("level", ""))
        else:
            print(f"State '{state_name}' not found.")
            
    def get_current_level(self):
        return self.current_level

    def get_selected_level(self):
        return self.selected_level
    
    def get_selected_character(self):
        print(f"StateManager: getting selected character ({self.selected_character})")
        return self.selected_character
    
    def set_difficulty(self, difficulty):
        self.difficulty = difficulty
    
    def get_difficulty(self):
        return self.difficulty
    
    def set_selected_character(self, character_index):
        print(f"StateManager: setting selected character to {character_index}")
        self.selected_character = character_index
        
    def update(self):
        if self.current_state:
            self.current_state.update()
            
    def draw(self, screen):
        if self.current_state:
            self.current_state.draw(screen)