Save player mode.

[naja.git] / naja / gameboard.py

diff --git a/naja/gameboard.py b/naja/gameboard.py
index 18ecaf5c62d369af2f8a3d32a0493b45b0926923..8a1daa7673b28d34dc1c264f5c5d0c3b3b3d5325 100644 (file)
--- a/naja/gameboard.py
+++ b/naja/gameboard.py
@@ -19,10 +19,10 @@ class GameBoard(object):
         self.health = state['health']
         self.wins = state['wins']
         self.locations = [item.copy() for item in state['locations']]
-        self.puzzle = state['puzzle']
+        self.puzzle = state.get('puzzle', False)
         self.player = player
         self.board_locations = board_locations
-        self.player_mode = EXAMINE
+        self.player_mode = state.get('player_mode', EXAMINE)
 
     @classmethod
     def new_game(cls, deck,
@@ -63,6 +63,7 @@ class GameBoard(object):
             'puzzle': self.puzzle,
             'player': self.player.export(),
             'board_locations': self.export_board_locations(),
+            'player_mode': self.player_mode,
         }
 
     @classmethod