ps9pr3.py

#
# Playing the game 
#   

from ps9pr1 import Board
from ps9pr2 import Player
import random
    
def connect_four(player1, player2):
    """ Plays a game of Connect Four between the two specified players,
        and returns the Board object as it looks at the end of the game.
        inputs: player1 and player2 are objects representing Connect Four
                  players (objects of the Player class or a subclass of Player).
                  One player should use 'X' checkers and the other should
                  use 'O' checkers.
    """
    # Make sure one player is 'X' and one player is 'O'.
    if player1.checker not in 'XO' or player2.checker not in 'XO' \
       or player1.checker == player2.checker:
        print('need one X player and one O player.')
        return None

    print('Welcome to Connect Four!')
    print()
    board = Board(6, 7)
    print(board)
    
    while True:
        if process_move(player1, board) == True:
            return board

        if process_move(player2, board) == True:
            return board


def process_move(player, board):
    """
    takes two parameters: a Player object for the player whose move is being processed,
    and a Board object for the game that is being played.
    """
    print(player.__repr__() + "'s" + " " + "turn")
    next_move = player.next_move(board)
    board.add_checker(player.checker,next_move)
    print()
    print(board)
    print()
    if board.is_win_for(player.checker) == True:
        print(player.__repr__() + " " + "wins in" + " " + str(player.num_moves) + " " + "moves.")
        print("Congratulations!")
        return True
    elif board.is_full() == True:
        print("It's a tie!")
        return True
    else:
        return False
   

class RandomPlayer(Player):
    def next_move(self, board):
        """
        overrides (i.e., replaces) the next_move method that is inherited from Player
        """
        lc = [x for x in range(board.width) if board.can_add_to(x) == True]
        mymove = random.choice(lc)
        self.num_moves += 1
        return mymove