naja/scenes/new_game.py

   1 """
   2 Load and save scenes.
   3 """
   4
   5 import pygame.locals as pgl
   6
   7 from naja.constants import KEYS
   8 from naja.events import SceneChangeEvent, LoadGameEvent
   9 from naja.gamestate import GameState
  10 from naja.scenes.scene import Scene
  11 from naja.widgets.selector import SelectorWidget
  12 from naja.widgets.text import TextWidget
  13
  14
  15 class NewGameScene(Scene):
  16     def __init__(self, state):
  17         super(NewGameScene, self).__init__(state)
  18         selector = SelectorWidget()
  19         self.add(selector)
  20
  21         easy = TextWidget((100, 100), 'Easy', fontsize=32, colour='white')
  22         easy.add_callback('click', self.easy_game)
  23         selector.add(easy)
  24
  25         standard = TextWidget(
  26             (100, 150), 'Standard', fontsize=32, colour='white')
  27         standard.add_callback('click', self.standard_game)
  28         selector.add(standard)
  29
  30         hard = TextWidget((100, 200), 'Hard', fontsize=32, colour='white')
  31         hard.add_callback('click', self.hard_game)
  32         selector.add(hard)
  33
  34     def easy_game(self, event):
  35         self.start_game(GameState.new(max_health=5, wins_required=3))
  36
  37     def standard_game(self, event):
  38         self.start_game(GameState.new(max_health=4, wins_required=4))
  39
  40     def hard_game(self, event):
  41         self.start_game(GameState.new(max_health=3, wins_required=5))
  42
  43     def start_game(self, state):
  44         from naja.scenes.game import GameScene
  45         LoadGameEvent.post(state=state)
  46         SceneChangeEvent.post(GameScene)
  47
  48     def handle_scene_event(self, ev):
  49         if ev.type == pgl.KEYDOWN and ev.key in KEYS.QUIT:
  50             from naja.scenes.menu import MenuScene
  51             SceneChangeEvent.post(MenuScene)
  52             return