Enable continue on save file existance

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

diff --git a/tabakrolletjie/scenes/menu.py b/tabakrolletjie/scenes/menu.py
index dcad16b3d27057898a2f80e6d448b6b06ce24d38..67336112c6c3e7a65151bfb524d23117bf1c7369 100644 (file)
--- a/tabakrolletjie/scenes/menu.py
+++ b/tabakrolletjie/scenes/menu.py
@@ -6,7 +6,8 @@ from .base import BaseScene
 from ..events import QuitEvent, SceneChangeEvent
 from ..loader import loader
 from ..constants import FONTS
-from ..widgets import TextButton
+from ..widgets import TextButton, SpacerButton
+from ..utils import save_file_exists
 
 
 class MenuScene(BaseScene):
@@ -17,11 +18,16 @@ class MenuScene(BaseScene):
         font_title = loader.load_font(FONTS['bold'], size=32)
         self._title = font_title.render('A Game with a title', True,
                                         (255, 255, 255))
+        self._cont_but = TextButton("Continue / Retry", (255, 255, 255),
+                                    name='continue')
+        if not save_file_exists():
+            self._cont_but.disable()
         self._menu = [
-            TextButton("Load Level", (255, 255, 255), name='load level'),
-            TextButton("Start Game (Day)", (255, 255, 255), name='start game'),
-            TextButton("Load Saved Game", (255, 255, 255), name='load game'),
+            TextButton("New Game", (255, 255, 255), name='new game'),
+            self._cont_but,
             TextButton("Help", (255, 255, 255), name='help'),
+            SpacerButton((50, 30)),
+            TextButton("Quit", (255, 255, 255), name='quit'),
         ]
 
     def render(self, surface, gamestate):
@@ -71,11 +77,13 @@ class MenuScene(BaseScene):
                 self._do_help()
         elif ev.type == pgl.MOUSEBUTTONDOWN:
             pressed = self._get_pressed(ev)
-            if pressed == 'load level':
+            if pressed == 'new game':
                 self._do_load_level()
-            elif pressed == 'start game':
+            elif pressed == 'continue':
                 self._do_day()
             elif pressed == 'help':
                 self._do_help()
+            elif pressed == 'quit':
+                QuitEvent.post()
             elif pressed:
                 print 'Pressed', pressed