the_resistance.py

from random import *
from spy import *
from resistance_member import *

# For each array that represents a 'mode' of the game (determined by
# the number of players) the first value of the array is the number
# of members, the second is the number ofspies and the following are 
# the amount of people on each mission
five_players = [5,2,2,3,2,3,3]
six_players = [6,2,2,3,4,3,4]
seven_players = [7,3,2,3,3,4,4]
eight_players = [8,3,3,4,4,5,5]
nine_players = [9,3,3,4,4,5,5]
ten_players = [10,4,3,4,4,5,5]

class Resistance_Game:

  def __init__(self, game_info):
    self.round_counts = []
    for i in range(5):
      self.round_counts.append(game_info[i + 2])

    # represents the leader in the self.players array, will increment
    # by one after every turn
    self.leader_index = 0

    self.round = 0          # the round the game state is currently in
    self.missions_won = 0   # missions won by the resistance
    self.missions_lost = 0  # missions lost by the resistance
    self.game_over = 0
    self.game_won = 0       # 0 if lost, 1 if won

    # The mission history is a dictionary of the round numbers 0-4 mapped
    # to the outcomes: 
    #    [[1, 4], [P, F]]
    #  where the first list is the list of members that go on the mission
    #  and the second list is the list of votes (does not correspond with the 
    #  ordering of the team list)
    self.mission_history = {}

    self.num_spies = game_info[1]
    self.num_players = game_info[0]

    # spy selection
    spy_indicies = []
    spy_indicies.append(randint(0, self.num_players - 1))
    for i in range(self.num_spies-1):
      while True:
        rand_index = randint(0, self.num_players - 1)
        if rand_index not in spy_indicies:
          break
      spy_indicies.append(rand_index)

    # making the list of players
    self.players = []
    for i in range(self.num_players):
      if i in spy_indicies:
        self.players.append(Spy(self.num_players, i, self.round_counts, spy_indicies))
      else:
        self.players.append(Resistance_Member(self.num_players, i, self.round_counts))

  def play_game(self):
    for i in range(5):
      if not self.game_over:
        self.play_round()
        self.round += 1
        self.leader_index += 1
        self.update_players()
      else:
        break

  def play_round(self):
    while True:
      # The leader selects the team going on the mission (list of indicies of the players in self.players)
      mission_team = self.players[self.leader_index].select_mission_team()

      # The players vote on the team
      vote_total = 0
      for player in self.players:
        vote_total += player.vote_on_mission(mission_team)

      if float(vote_total)/float(self.num_players) > 0.5:
        # mission success
        break
      else:
        # mission fails, the leadership changes and we repeat this process
        self.leader_index += 1

    # The mission team will go on the mission and either pass or fail the mission
    mission_results = []
    for i in range(self.num_players):
      if i in mission_team:
        mission_results.append(self.players[i].go_on_mission(mission_team))
    shuffle(mission_results)

    # add the results to the mission history
    self.mission_history[self.round] = [mission_team, mission_results]

    # Determine the next state based on the result of this round
    if 'F' in mission_results:
      self.missions_lost += 1
      if self.missions_lost >= 3:
        self.end_of_game()
    else:
      self.missions_won += 1
      if self.missions_won >= 3:
        self.end_of_game()

  def end_of_game(self):
    if self.missions_won >= 3 or self.missions_lost >= 3:
      self.game_over = 1
      # for i in range(self.round + 1):
      #   print self.mission_history[i]
      if self.missions_won >= 3:
        self.game_won = 1
        # print "The Resistance WON!"
      # else:
        # print "The Reistance has been thwarted, the Spies WON!"
    else:
      print "A problem has occured: end state has been reached, but neither team has won :("

  def update_players(self):
    for player in self.players:
      player.round = self.round
      player.missions_lost = self.missions_lost
      player.mission_history = self.mission_history

# game = Resistance_Game(six_players)
# game.play_game()