tictactoe.py

import random


class TicTacToe:
    def __init__(self, size=3):
        self.size = size
        self.grid = {i: {j: " " for j in range(size)} for i in range(size)}
        self.turn = 0  # 0 for player 1 (X) and 1 for player 2 (O)

    def place(self, x, y):
        if x < 0 or x >= self.size or y < 0 or y >= self.size:  # Check grid boundaries
            return False
        if self.grid[x][y] != " ":  # Check if cell is already occupied
            return False
        # Determine the piece based on whose turn it is
        piece = "X" if self.turn == 0 else "O"
        # Place the piece
        self.grid[x][y] = piece
        # Check for a win after the piece is placed
        if self.check_winner(x, y, piece):
            self.print_board()  # Print the final winning board
            winner = self.turn  # Capture the winner before toggling the turn
            self.reset_game()  # Optional: reset game after a win
            return winner
        # Check for Cat's Game (draw)
        if self.check_draw():
            self.print_board()  # Print the final board
            self.reset_game()  # Optional: reset game after a draw
            return "Draw"
        # Toggle the turn
        self.turn = 1 - self.turn
        self.print_board()  # Print board after every move
        return None  # Indicate ongoing game

    def check_winner(self, x, y, piece):
        # Check the row and column of the last move for a win
        if all(self.grid[x][j] == piece for j in range(self.size)) or all(self.grid[i][y] == piece for i in range(self.size)):
            return True
        # Check diagonals only if x and y are on a diagonal
        # Check the main diagonal
        if x == y and all(self.grid[i][i] == piece for i in range(self.size)):
            return True
        # Check the anti-diagonal
        if x + y == self.size - 1 and all(self.grid[i][self.size-i-1] == piece for i in range(self.size)):
            return True
        return False

    def check_draw(self):
        # Check if all cells are filled
        return all(self.grid[i][j] != " " for i in range(self.size) for j in range(self.size))

    def reset_game(self):
       # Resets the game to the initial state
        self.grid = {i: {j: " " for j in range(
            self.size)} for i in range(self.size)}
        self.turn = 0  # Optionally reset the turn to player 0

    def print_board(self):
        for row in range(self.size):
            print('|'.join(self.grid[row].values()))
            if row < self.size - 1:
                print("-" * (self.size * 2 - 1))  # Print a line between rows


class AdvancedTicTacToe:
    def __init__(self, size=3, win_condition=None):
        self.size = size
        # Defaults to size for classic game
        self.win_condition = win_condition if win_condition else size
        self.grid = [[" " for _ in range(size)] for _ in range(size)]
        self.turn = "X"  # Start with player X
        self.winner = None
        self.moves_made = 0
        # Could be 'Human' or 'Computer'
        self.players = {"X": "Human", "O": "Human"}

    def set_player_mode(self, player_x="Human", player_o="Human"):
        """Set whether players are human or computer."""
        self.players["X"] = player_x
        self.players["O"] = player_o

    def play_game(self):
        """Main game loop."""
        while not self.winner and self.moves_made < self.size ** 2:
            self.print_board()
            if self.players[self.turn] == "Human":
                self.human_move()
            else:
                self.computer_move()
            if self.check_winner():
                self.winner = self.turn
                print(f"Player {self.turn} wins!")
            elif self.moves_made == self.size ** 2:
                print("It's a draw!")
            self.turn = "O" if self.turn == "X" else "X"
        self.print_board()

    def human_move(self):
        valid_move = False
        while not valid_move:
            try:
                x, y = map(int, input(
                    f"Player {self.turn}, enter your move as 'row col': ").split())
                if 0 <= x < self.size and 0 <= y < self.size and self.grid[x][y] == " ":
                    self.grid[x][y] = self.turn
                    self.moves_made += 1
                    valid_move = True
                else:
                    print("Invalid move. Try again.")
            except (ValueError, IndexError):
                print("Please enter row and column as two numbers separated by a space.")

    def computer_move(self):
        print(f"Player {self.turn}'s (Computer) turn:")
        empty_cells = [(i, j) for i in range(self.size)
                       for j in range(self.size) if self.grid[i][j] == " "]
        x, y = random.choice(empty_cells)
        self.grid[x][y] = self.turn
        self.moves_made += 1

    def check_winner(self):
        # Check rows and columns
        for i in range(self.size):
            if all(self.grid[i][j] == self.turn for j in range(self.size)) or all(self.grid[j][i] == self.turn for j in range(self.size)):
                return True
        # Check diagonals
        if all(self.grid[i][i] == self.turn for i in range(self.size)) or all(self.grid[i][self.size-i-1] == self.turn for i in range(self.size)):
            return True
        return False

    def print_board(self):
        print("\n" + "\n".join([" | ".join(row) for row in self.grid]))
        print()

    def reset_game(self):
        """Reset the game to start a new round."""
        self.grid = [[" " for _ in range(self.size)] for _ in range(self.size)]
        self.turn = "X"
        self.winner = None
        self.moves_made = 0