Add quitting.

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

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index 3c703c3990d4d7e68c4d988e383da31888012a1a..822c87a1eb77db35d98b965694c3aff81e7b4dae 100644 (file)
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -2,17 +2,21 @@
 Main menu scene.
 """
 
+import pygame.locals as pgl
+
 from naja.scenes.scene import Scene
 from naja.widgets.text import TextWidget
+from naja.events import QuitGameEvent
 
 
 class MenuScene(Scene):
 
     def __init__(self):
-        self.tw = TextWidget((10, 10), 'Haai', colour=(255, 255, 255))
-
-    def render(self, surface):
-        self.tw.draw(surface)
+        super(MenuScene, self).__init__()
+        self.widgets.append(TextWidget((10, 10), 'Haai',
+                            colour=(255, 255, 255)))
 
     def handle_event(self, ev):
-        pass
+        if ev.type == pgl.KEYUP and ev.key in (pgl.K_q, pgl.K_ESCAPE):
+            QuitGameEvent.post()
+            return