from naja.constants import(
BITS, DIRECTION_BITS, CONDITION_BITS, PLAYER_DEFAULTS,
- ACT, EXAMINE, ROTATION)
+ ROTATION)
from naja.options import options
from naja.player import Player
from naja import actions
self.puzzle = state.get('puzzle', False)
self.player = player
self.board_locations = board_locations
- self.player_mode = state.get('player_mode', EXAMINE)
self.has_cheated = state.get('cheater', options.cheat_enabled)
self.clock_count = state.get('clock_count', 0)
self.replacement_params = state.get('replacement_params', None)
'puzzle': self.puzzle,
'player': self.player.export(),
'board_locations': self.export_board_locations(),
- 'player_mode': self.player_mode,
'clock_count': self.clock_count,
'replacement_params': self.replacement_params,
}
# Find which cards are at their maximum and exclude them from
# the choice list
counts = {}
- choices = {card['card_name']: card for card in cards}
+ choices = dict((card['card_name'], card) for card in cards)
for pos, card in board_locations:
if pos == position:
# skip the card we're replacing if appropriate
del choices[key]
return choice(choices.values())
- def shift_location_row(self, change, is_vertical):
+ def shift_location_row(self, change, is_vertical, skip_player=True):
px, py = self.player.position
shifted_locations = {}
mkpos = lambda i: (px, i) if is_vertical else (i, py)
for i in range(5):
- if (px, py) == mkpos(i):
+ if skip_player and (px, py) == mkpos(i):
continue
new_i = (i + change) % 5
- if (px, py) == mkpos(new_i):
+ if skip_player and (px, py) == mkpos(new_i):
new_i = (new_i + change) % 5
shifted_locations[mkpos(new_i)] = self.board_locations[mkpos(i)]
self.board_locations.update(shifted_locations)
- def shift_locations(self, direction):
+ def shift_locations(self, direction, skip_player=True):
if BITS[direction] == BITS.NORTH:
- self.shift_location_row(-1, is_vertical=True)
+ self.shift_location_row(-1, is_vertical=True,
+ skip_player=skip_player)
elif BITS[direction] == BITS.SOUTH:
- self.shift_location_row(1, is_vertical=True)
+ self.shift_location_row(1, is_vertical=True,
+ skip_player=skip_player)
elif BITS[direction] == BITS.EAST:
- self.shift_location_row(1, is_vertical=False)
+ self.shift_location_row(1, is_vertical=False,
+ skip_player=skip_player)
elif BITS[direction] == BITS.WEST:
- self.shift_location_row(-1, is_vertical=False)
+ self.shift_location_row(-1, is_vertical=False,
+ skip_player=skip_player)
def rotate_locations(self, direction):
px, py = self.player.position
def allow_chess_move(self, chesspiece):
self.player.allow_chess_move(chesspiece)
- def change_mode(self, new_mode):
- """Advance to the next mode"""
- if new_mode == self.player_mode:
- raise RuntimeError("Inconsistent state. Setting mode %s to itself"
- % self.player_mode)
- elif new_mode in (ACT, EXAMINE):
- self.player_mode = new_mode
- if new_mode is EXAMINE:
- self.board_update()
- else:
- raise RuntimeError("Illegal player mode %s" % self.player_mode)
-
def board_update(self):
self.clock_count += 1
for position, location in self.board_locations.iteritems():