Beispiel #1
0
class ChessGame(object):

    """A chess game manager"""

    def __init__(self):
        self.white, self.black = Player(), Player()
        self.chessboard = Chessboard()
        self.chessboard.add_players(self.white, self.black)
        self.greet()

    def greet(self):
        """Greet the players and begin game."""
        print("Welcome to KingSnake.\n"
              "Enter moves directly or press Enter to enter the menu.")
        self.get_move()

    def get_move(self, message=""):
        """Show board and prompt for current move, then execute it."""
        valid_move = False
        while not valid_move:
            self.show(message)
            move = raw_input("Please enter your move (e.g. E2 E4) or enter to "
                             "access the menu: ")
            if not move:
                self.menu()
            else:
                try:
                    start_position, end_position = move.split()
                    self.chessboard.current_player.move(start_position.upper(),
                                                        end_position.upper())
                    valid_move = True
                except ValueError:
                    message = ("Please enter your move in the following form:"
                               "\n{start position} {end position}\n"
                              "Positions are notated using their letter "
                              "followed by their number.\n"
                              "Example valid move request to move from A1 to "
                              "A2: 'A1 A2'")
                except KeyError:
                    message = "Only valid fields are allowed."
                except ChessError as error:
                    message = error
        self.get_move(message)

    def menu(self):
        """Allow user to do something other than move pieces"""

        def quit_game():
            """Quit game."""
            sys.exit()

        def resign():
            """Resign."""
            print("{} resigns. {} is the winner!".format(
                 self.chessboard.current_player.color.capitalize(),
                 self.chessboard.current_player.opponent.color.capitalize()))
            time.sleep(10)
            quit_game()

        def restart():
            """Start new game."""
            self.__init__()

        def save_game():
            """Save game to file."""
            file_name = raw_input("What file would you like to save to?: ")
            try:
                with open(file_name, "w") as saved_game:
                    pickle.dump(self.chessboard, saved_game)
            except IOError:
                self.get_move("The file you have chosen is invalid. "
                              "Please enter a valid filename.")
            self.get_move("Game saved to {}.".format(file_name))

        def load_game():
            """Load game from file."""
            file_name = raw_input("What file would you like to load from?: ")
            try:
                with open(file_name) as saved_game:
                    self.chessboard = pickle.load(saved_game)
                    self.white = self.chessboard.players["white"]
                    self.black = self.chessboard.players["black"]
            except IOError:
                self.get_move("The file you have chosen is invalid. "
                              "Please enter a valid filename.")

        def undo_turn():
            """Undo turn."""
            self.chessboard.rollback()
            self.get_move("Move restored.")

        def return_to_game():
            """Resume play."""
            self.get_move()

        menu_choices = []
        for function in (quit_game, resign, restart, save_game, load_game,
                         undo_turn, return_to_game):
            menu_choices.append((function, function.__doc__))
        valid_choice = False
        while not valid_choice:
            for number, choice in enumerate(menu_choices):
                print("{}. {}".format(number + 1, choice[1]))
            try:
                decision = int(raw_input("What would you like to do?: ")) - 1
                try:
                    valid_choice = True
                    menu_choices[decision][0]()
                except IndexError:
                    valid_choice = False
                    print("Please enter a valid menu number.")
            except (ValueError, IndexError):
                print("Please enter a valid menu number.")

    def show(self, message=""):
        """Show chessboard and print current player."""
        print(self.chessboard)
        print("{}\n"
              "It's {}'s turn.".format(message,
                                       self.chessboard.current_player.color))