five_in_a_row.py

import tkinter as tk

root = tk.Tk()
root.title("Five in a Row / Caro Game")

WIDTH = root.winfo_screenwidth()
HEIGHT = root.winfo_screenheight()
CELL = 50
ROWS = HEIGHT // CELL
COLS = WIDTH // CELL
TURN = "X"
WINNING_CONDITION = 3  # Number of wins to end the game

# Board state
board = [[None for _ in range(COLS)] for _ in range(ROWS)]

# Winners state
winners = []

canvas = tk.Canvas(root, width=WIDTH, height=HEIGHT)
canvas.pack()


def draw_board():
    for i in range(COLS):
        canvas.create_line(0, CELL * i, WIDTH, CELL * i, fill="black")  # Horizontal lines
        canvas.create_line(CELL * i, 0, CELL * i, HEIGHT, fill="black")  # Vertical lines


def play(event):
    global TURN, winners
    col = event.x // CELL
    row = event.y // CELL
    x_center = col * CELL + CELL // 2
    y_center = row * CELL + CELL // 2
    if board[row][col] is not None:
        return
    board[row][col] = TURN
    canvas.create_text(x_center, y_center, text=TURN, font=("Arial", 20))
    TURN = "X" if TURN == "O" else "O"

    # Check if there is a winner
    winner = detect_winner()
    if winner is not None:
        TURN = "X" if winner == "O" else "O"  # Switch turn based on last winner
        winners.append(winner)
        canvas.create_text(100, 100, text=f"{winner} wins!", font=("Arial", 50), anchor="w")

        if winners.count(winner) == WINNING_CONDITION:
            # End the game and display the result
            x_wins = winners.count("X")
            o_wins = winners.count("O")
            canvas.create_text(
                100, 200, text=f"{winner} wins the game!", font=("Arial", 50), anchor="w"
            )
            canvas.create_text(
                100, 300, text=f"Score: {x_wins} - {o_wins}", font=("Arial", 30), anchor="w"
            )
            canvas.unbind("<Button-1>")
            canvas.bind("<Button-1>", lambda _: root.quit())
            return
        canvas.unbind("<Button-1>")
        canvas.bind("<Button-1>", restart)
        return


# Helper function to check if the winner is blocked
def is_blocked(row, col, current) -> bool:
    if row < 0 or row > ROWS or col < 0 or col > COLS:
        return False
    return board[row][col] is not None and board[row][col] != current


def detect_winner() -> str | None:
    for row in range(ROWS):
        for col in range(COLS):
            current = board[row][col]
            if current is None:
                continue
            # Check horizontal
            if col + 4 < COLS and all(board[row][col + i] == current for i in range(5)):
                if is_blocked(row, col - 1, current) and is_blocked(row, col + 5, current):
                    continue
                return current
            # Check vertical
            if row + 4 < ROWS and all(board[row + i][col] == current for i in range(5)):
                if is_blocked(row - 1, col, current) and is_blocked(row + 5, col, current):
                    continue
                return current
            # Check diagonal from left to right
            if (
                row + 4 < ROWS
                and col + 4 < COLS
                and all(board[row + i][col + i] == current for i in range(5))
            ):
                if is_blocked(row - 1, col - 1, current) and is_blocked(row + 5, col + 5, current):
                    continue
                return current
            # Check diagonal from right to left
            if (
                row + 4 < ROWS
                and col - 4 >= 0
                and all(board[row + i][col - i] == current for i in range(5))
            ):
                if is_blocked(row - 1, col + 1, current) and is_blocked(row + 5, col - 5, current):
                    continue
                return current
    return None


def restart(_event):
    global board, TURN
    board = [[None for _ in range(COLS)] for _ in range(ROWS)]
    canvas.delete("all")
    draw_board()
    canvas.unbind("<Button-1>")
    canvas.bind("<Button-1>", play)


draw_board()
canvas.bind("<Button-1>", play)
root.mainloop()