Add missing state arguments to scene __init__s.

[naja.git] / naja / scenes / menu.py

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index 96c8c7a5677e8374458a1f2d1694d0209f6d274a..9a0fe71318cb3095dd69b16b20c7f69ea8890599 100644 (file)
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -10,18 +10,22 @@ from naja.events import QuitGameEvent
 from naja.scenes.credits import CreditsScene
 from naja.events import SceneChangeEvent
 
+
 class MenuScene(Scene):
 
-    def __init__(self):
-        super(MenuScene, self).__init__()
-        self.widgets.append(TextWidget((10, 10), 'Haai', fontsize=32,
+    def __init__(self, state):
+        super(MenuScene, self).__init__(state)
+        self.widgets.append(TextWidget((10, 10), 'Game', fontsize=32,
+                            colour='white'))
+        self.widgets.append(TextWidget((10, 40), 'Credits', fontsize=32,
+                            colour='white'))
+        self.widgets.append(TextWidget((10, 70), 'Quit', fontsize=32,
                             colour='white'))
 
     def handle_event(self, ev):
-        if ev.type == pgl.KEYUP and ev.key in (pgl.K_q, pgl.K_ESCAPE):
+        if ev.type == pgl.KEYDOWN and ev.key in (pgl.K_q, pgl.K_ESCAPE):
             QuitGameEvent.post()
             return
-        if ev.type == pgl.KEYUP and ev.key is pgl.K_RETURN:
-            credits_menu = CreditsScene()
-            SceneChangeEvent.post(credits_menu)
+        if ev.type == pgl.KEYDOWN and ev.key is pgl.K_RETURN:
+            SceneChangeEvent.post(CreditsScene)
             return