board.py

class Board(object):
    """棋盘类"""
    def __init__(self):
        self.board_data = [" "] * 9 # 棋盘数据
        self.movable_list = list(range(9)) # 可移动列表
    def show_board(self, show_index = False):
        """显示棋盘
        :param show_index: True 表示显示索引 / False 表示显示数据
        """
        for i in (0, 3, 6):
            print("     |     |") # 5个空格
            if show_index:
                print("  %d  |  %d  |  %d" % (i, i + 1, i +2)) # 2个空格
            else:
                print("  %s  |  %s  |  %s" % (self.board_data[i], self.board_data[i + 1], self.board_data[i + 2]))
            print("     |     |")
            if i != 6:
                print("-" * 23)
    def move_down(self, index, chess):
        """在指定位置落子
        :param index: 列表索引
        :param chess: 棋子类型 x 或 o
        """
        # 1. 判断 index 是否在可移动列表中
        if index not in self.movable_list:
            print("%d 位置不允许落子" % index)
            return
        # 2.修改棋盘数据
        self.board_data[index] = chess
        # 3. 修改可移动列表
        self.movable_list.remove(index)
    def is_draw(self):
        """是否平局"""
        return len(self.movable_list) == 0
    def is_win(self, chess, ai_index = -1):
        """是否胜利
        :param chess: 玩家的棋子
        :param ai_index: 预判索引， -1 直接判断当前棋盘数据
        """
        # 1. 定义检查方向列表
        check_dirs = [[0, 1, 2], [3, 4, 5], [6, 7, 8], [0, 3, 6], [1, 4, 7], [2, 5, 8], [0, 4, 8], [2, 4, 6]]
        # 2. 定义局部变量记录棋盘数据副本
        data = self.board_data.copy()
        #  判断是否预判胜利
        if ai_index > 0:
            data[ai_index] = chess
        # 3. 遍历检查方向列表是否判断胜利
        for item in check_dirs:
            if (data[item[0]] == chess and data[item[1]] == chess and data[item[2]] == chess):
                return True
        return False
    def reset_board(self):
        """重置棋盘"""
        # 1. 清空可移动数据列表数据
        self.movable_list.clear()
        # 2. 重置数据
        for i in range(9):
            self.board_data[i] = " "
            self.movable_list.append(i)

if __name__ == '__main__':
    # 1. 测试初始化
    board = Board()
    print(board.board_data)
    print(board.movable_list)
    # 2. 显示棋盘
    print("--- 显示棋盘" + "-" * 50)
    board.show_board(True) # 显示索引
    print("-" * 50)
    board.show_board() # 显示数据
    # 3. 测试落子
    # print("--- 测试落子" + "-" * 50)
    # board.move_down(0, "X")
    # board.show_board()
    # print(board.movable_list)
    # board.move_down(0, "X") # 测试在已有棋子的位置落子
    # 4. 判断平局
    print("--- 判断平局" + "-" * 50)
    print("是否平局 %d" % board.is_draw())
    # board.movable_list.clear() # 清空可移动索引列表
    # print("是否平局 %d" % board.is_draw())
    # 5. 判断胜利
    print("--- 判断胜利" + "-" * 50)
    print("是否胜利 %d" % board.is_win("X"))
    board.move_down(0, "X")
    board.move_down(1, "X")
    board.move_down(2, "X")
    board.show_board()
    print("是否胜利 %d" % board.is_win("X"))
    # 6. 测试重置棋盘
    print("--- 重置棋盘" + "-" * 50)
    board.reset_board()
    board.show_board()
    print(board.movable_list)