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 e87005491f9fd79867b1fbc30b5c9242f298a08c..67336112c6c3e7a65151bfb524d23117bf1c7369 100644 (file)
--- a/tabakrolletjie/scenes/menu.py
+++ b/tabakrolletjie/scenes/menu.py
@@ -6,22 +6,29 @@ 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):
     def enter(self, gamestate):
         if gamestate.station is None:
             print "Loading Station Alpha ..."
-            gamestate.load_station("station-alpha.json")
+            gamestate.set_station(loader.load_station("station-alpha.json"))
         font_title = loader.load_font(FONTS['bold'], size=32)
         self._title = font_title.render('A Game with a title', True,
                                         (255, 255, 255))
-        self._menu = {
-            'load level': TextButton("Load Level", (255, 255, 255)),
-            'saved game': TextButton("Load Saved Game", (255, 255, 255)),
-            'start game': TextButton("Start Game (Day)", (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("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):
         surface.fill((0, 128, 0))
@@ -30,8 +37,7 @@ class MenuScene(BaseScene):
         surface.blit(self._title, pos, None)
 
         height = 150
-        for label in sorted(self._menu):
-            item = self._menu[label]
+        for item in self._menu:
             if not item.position:
                 item.position = ((surface.get_width() - item.get_width()) / 2,
                                  height)
@@ -39,15 +45,23 @@ class MenuScene(BaseScene):
             height += item.get_height() + 30
 
     def _get_pressed(self, ev):
-        for label, button in self._menu.items():
+        for button in self._menu:
             if button.pressed(ev):
-                return label
+                return button.name
         return None
 
     def _do_day(self):
         from .day import DayScene
         SceneChangeEvent.post(scene=DayScene())
 
+    def _do_load_level(self):
+        from .load_level import LoadLevelScene
+        SceneChangeEvent.post(scene=LoadLevelScene())
+
+    def _do_help(self):
+        from .help import HelpScene
+        SceneChangeEvent.post(scene=HelpScene())
+
     def event(self, ev, gamestate):
         if ev.type == pgl.KEYDOWN:
             if ev.key in (pgl.K_q, pgl.K_ESCAPE):
@@ -58,14 +72,18 @@ class MenuScene(BaseScene):
             elif ev.key == pgl.K_d:
                 self._do_day()
             elif ev.key == pgl.K_l:
-                print "Loading Station Alpha ..."
-                gamestate.load_station("station-alpha.json")
+                self._do_load_level()
+            elif ev.key == pgl.K_h:
+                self._do_help()
         elif ev.type == pgl.MOUSEBUTTONDOWN:
             pressed = self._get_pressed(ev)
-            if pressed == 'load level':
-                print "Loading Station Alpha ..."
-                gamestate.load_station("station-alpha.json")
-            elif pressed == 'start game':
+            if pressed == 'new game':
+                self._do_load_level()
+            elif pressed == 'continue':
                 self._do_day()
+            elif pressed == 'help':
+                self._do_help()
+            elif pressed == 'quit':
+                QuitEvent.post()
             elif pressed:
                 print 'Pressed', pressed