"""Where a level object hides."""

from pygame.locals import KEYDOWN

from mamba.constants import ESCAPE_KEYS
from mamba.engine import Habitat, NewHabitatEvent
from mamba.widgets.level import LevelWidget
from mamba.level import Level


class LevelHabitat(Habitat):
    def __init__(self, level_name):
        super(LevelHabitat, self).__init__()
        self.level = Level(level_name)
        self.level_widget = LevelWidget(self.level)
        self.container.add(self.level_widget)
        self.container.add_callback(KEYDOWN, self.keydown_event)

    def keydown_event(self, ev, widget):
        if ev.key in ESCAPE_KEYS:
            from mamba.habitats.mainmenu import MainMenu
            NewHabitatEvent.post(MainMenu())