X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fscenes%2Fmenu.py;h=91f858b0d6304ef62e55e6142c40e9737b66d5f0;hb=73518c3b0cec7d960474fc9ab0a8c30a1cd41ec7;hp=317e50f4886949601f2512c2bc78080a40c916f5;hpb=bcc041da95b01ae36c2198b5c6bc4bd75dcc0c24;p=naja.git

diff --git a/naja/scenes/menu.py b/naja/scenes/menu.py
index 317e50f..91f858b 100644
--- a/naja/scenes/menu.py
+++ b/naja/scenes/menu.py
@@ -4,8 +4,10 @@ Main menu scene.
 
 import pygame.locals as pgl
 
+from naja.constants import KEYS
 from naja.scenes.scene import Scene
 from naja.widgets.text import TextWidget
+from naja.widgets.selector import SelectorWidget
 from naja.events import QuitGameEvent
 from naja.scenes.credits import CreditsScene
 from naja.scenes.game import GameScene
@@ -15,20 +17,22 @@ from naja.events import SceneChangeEvent
 class MenuScene(Scene):
     def __init__(self, state):
         super(MenuScene, self).__init__(state)
+        selector = SelectorWidget()
+        self.add(selector)
         run_game = TextWidget((100, 100), 'Game', fontsize=32, colour='white')
         run_game.add_callback('click',
                               lambda event: SceneChangeEvent.post(GameScene))
-        self.add(run_game)
+        selector.add(run_game)
         credits = TextWidget((100, 200), 'Credits', fontsize=32, colour='white')
         credits.add_callback('click',
                              lambda event: SceneChangeEvent.post(CreditsScene))
-        self.add(credits)
+        selector.add(credits)
         quit = TextWidget((100, 300), 'Quit', fontsize=32, colour='white')
         quit.add_callback('click', lambda event: QuitGameEvent.post())
-        self.add(quit)
+        selector.add(quit)
 
     def handle_scene_event(self, ev):
         if ev.type == pgl.KEYDOWN:
-            if ev.key in (pgl.K_q, pgl.K_ESCAPE):
+            if ev.key in KEYS.QUIT:
                 QuitGameEvent.post()
                 return