Merge branch 'master' of ctpug.org.za:koperkapel
[koperkapel.git] / koperkapel / scenes / menu.py
1 """ Main menu scene. """
2
3 from pgzero.constants import keys
4 from ..actors.buttons import TextButton
5 from ..actors.navigator import ActorNavigator
6 from .base import Scene, ChangeSceneEvent, QuitEvent
7
8
9
10
11 class MenuScene(Scene):
12     """ Main menu scene. """
13
14     def __init__(self):
15         super().__init__()
16         self._title = self.actors.add(TextButton("Main Menu"))
17         self._title.pos = (300, 200)
18         self._nav = ActorNavigator()
19         self._menu = [
20             TextButton("Play", action=self.change_to_level),
21             TextButton("Credits", action=self.change_to_credits),
22             TextButton("Quit", action=self.quit),
23         ]
24         for i, item in enumerate(self._menu):
25             item.pos = (300, 300 + (i * 100))
26             self.actors.add(item)
27         self._nav.connect(
28             self._menu, [keys.DOWN, keys.RIGHT], [keys.UP, keys.LEFT],
29             wrap=True)
30         self._nav.current.select()
31
32     def change_to_level(self):
33         from .level import LevelScene
34         return [ChangeSceneEvent(LevelScene('level1'))]
35
36     def change_to_credits(self):
37         from .credits import CreditsScene
38         return [ChangeSceneEvent(CreditsScene())]
39
40     def quit(self):
41         return [QuitEvent()]
42
43     def on_key_down(self, key, mod, unicode):
44         if key == keys.RETURN:
45             current = self._nav.current
46             if current is not None:
47                 return current.action()
48         else:
49             prev, current = self._nav.follow(key)
50             if prev is not None:
51                 prev.deselect()
52             if current is not None:
53                 current.select()