Морской бой

import tkinter as tk
import random
 
GRID_SIZE = 15  
SHIP_TYPES = {
    4: 1,
    3: 2,
    2: 3,
    1: 4
}
WATER = "~"
SHIP = "S"
HIT = "X"
MISS = "O"
BACKGROUND_COLOR = "#87CEEB"  
SHIP_COLOR = "#FFD700"  
HIT_COLOR = "#FF0000"  
MISS_COLOR = "#0000FF" 
 
player_grid = None
computer_grid = None
buttons_player = []
buttons_computer = []
player_ships = 0
computer_ships = 0
turns_left = 50
computer_targets = []
 
def create_grid():
    return [[WATER for _ in range(GRID_SIZE)] for _ in range(GRID_SIZE)]
 
def can_place_ship(grid, x, y, size, direction):
    dx, dy = (1, 0) if direction == "H" else (0, 1)
    for i in range(size):
        nx, ny = x + i * dx, y + i * dy
        if not (0 <= nx < GRID_SIZE and 0 <= ny < GRID_SIZE) or grid[nx][ny] != WATER:
            return False
        for adj_x in range(nx - 1, nx + 2):
            for adj_y in range(ny - 1, ny + 2):
                if 0 <= adj_x < GRID_SIZE and 0 <= adj_y < GRID_SIZE and grid[adj_x][adj_y] == SHIP:
                    return False
    return True
 
def place_ship(grid, x, y, size, direction):
    dx, dy = (1, 0) if direction == "H" else (0, 1)
    for i in range(size):
        nx, ny = x + i * dx, y + i * dy
        grid[nx][ny] = SHIP
 
def place_ships_automatically(grid):
    total_ships = 0
    for size, count in SHIP_TYPES.items():
        for _ in range(count):
            while True:
                x = random.randint(0, GRID_SIZE - 1)
                y = random.randint(0, GRID_SIZE - 1)
                direction = random.choice(["H", "V"])
                if can_place_ship(grid, x, y, size, direction):
                    place_ship(grid, x, y, size, direction)
                    total_ships += 1
                    break
    return total_ships
 
def place_player_ship(x, y):
    global player_ships
    if player_grid[x][y] == WATER and player_ships > 0:
        player_grid[x][y] = SHIP
        buttons_player[x][y].config(bg=SHIP_COLOR, state="disabled")
        player_ships -= 1
        status_label.config(text=f"Ваши корабли: {player_ships}")
        if player_ships == 0:
            status_label.config(text="Ваши корабли расставлены. Теперь стреляйте!")
            disable_player_buttons()
 
def disable_player_buttons():
    for row in buttons_player:
        for btn in row:
            btn.config(state="disabled")
 
def reveal_computer_ships():
    for x in range(GRID_SIZE):
        for y in range(GRID_SIZE):
            if computer_grid[x][y] == SHIP:
                buttons_computer[x][y].config(bg=SHIP_COLOR)
 
def disable_computer_buttons():
    for row in buttons_computer:
        for btn in row:
            btn.config(state="disabled")
 
def handle_click(x, y):
    global computer_ships, turns_left
 
    if computer_grid[x][y] == SHIP:
        computer_grid[x][y] = HIT
        buttons_computer[x][y].config(bg=HIT_COLOR, state="disabled")
        computer_ships -= 1
        status_label.config(text="Попадание!")
    elif computer_grid[x][y] == WATER:
        computer_grid[x][y] = MISS
        buttons_computer[x][y].config(bg=MISS_COLOR, state="disabled")
        status_label.config(text="Мимо!")
 
    turns_left -= 1
    turns_label.config(text=f"Ходов осталось: {turns_left}")
 
    if computer_ships == 0:
        status_label.config(text="Вы победили!")
        disable_computer_buttons()
        return
 
    if turns_left == 0:
        status_label.config(text="Игра окончена! Вы проиграли.")
        reveal_computer_ships()
        disable_computer_buttons()
        return
 
    computer_turn()
 
def computer_turn():
    global computer_targets, player_grid, buttons_player, player_ships
 
    if computer_targets:
        x, y = computer_targets.pop(0)
    else:
        x, y = random.randint(0, GRID_SIZE - 1), random.randint(0, GRID_SIZE - 1)
 
    while player_grid[x][y] in [HIT, MISS]:
        if computer_targets:
            x, y = computer_targets.pop(0)
        else:
            x, y = random.randint(0, GRID_SIZE - 1), random.randint(0, GRID_SIZE - 1)
 
    if player_grid[x][y] == SHIP:
        player_grid[x][y] = HIT
        buttons_player[x][y].config(bg=HIT_COLOR)
        player_ships -= 1
        add_adjacent_targets(x, y)
    else:
        player_grid[x][y] = MISS
        buttons_player[x][y].config(bg=MISS_COLOR)
 
    if player_ships == 0:
        status_label.config(text="Компьютер победил!")
        disable_computer_buttons()
 
def add_adjacent_targets(x, y):
    global computer_targets
    directions = [(-1, 0), (1, 0), (0, -1), (0, 1)]
    for dx, dy in directions:
        nx, ny = x + dx, y + dy
        if 0 <= nx < GRID_SIZE and 0 <= ny < GRID_SIZE and player_grid[nx][ny] not in [HIT, MISS]:
            computer_targets.append((nx, ny))
 
def start_game():
    global player_grid, computer_grid, buttons_player, buttons_computer
    global player_ships, computer_ships, turns_left, status_label, turns_label
 
    player_grid = create_grid()
    computer_grid = create_grid()
 
    player_ships = sum(SHIP_TYPES.values())
    computer_ships = place_ships_automatically(computer_grid)
 
    root = tk.Tk()
    root.title("Морской бой")
 
    def start_new_game():
        start_game()
        start_window.withdraw()
 
    # Стартовое окно
    start_window = tk.Toplevel(root)
    start_window.title("Морской бой - Стартовое меню")
    tk.Label(start_window, text="Добро пожаловать в Морской бой!", font=("Arial", 16)).grid(row=0, column=0, pady=10)
    tk.Button(start_window, text="Начать игру", font=("Arial", 12), command=start_new_game).grid(row=1, column=0, pady=10)
 
    # Поле игрока
    tk.Label(root, text="Ваше поле", font=("Arial", 14)).grid(row=0, column=0, padx=10, pady=10)
    player_frame = tk.Frame(root)
    player_frame.grid(row=1, column=0, padx=10, pady=10)  # Отступы для улучшения внешнего вида
    global buttons_player
    buttons_player = [
        [tk.Button(player_frame, width=4, height=2, bg=BACKGROUND_COLOR, relief="solid", 
                   command=lambda x=x, y=y: place_player_ship(x, y))
         for y in range(GRID_SIZE)] for x in range(GRID_SIZE)]
    for x, row in enumerate(buttons_player):
        for y, btn in enumerate(row):
            btn.grid(row=x, column=y)
 
    # Поле компьютера
    tk.Label(root, text="Поле компьютера", font=("Arial", 14)).grid(row=0, column=1, padx=10, pady=10)
    computer_frame = tk.Frame(root)
    computer_frame.grid(row=1, column=1, padx=10, pady=10)  # Отступы для улучшения внешнего вида
    global buttons_computer
    buttons_computer = [
        [tk.Button(computer_frame, width=4, height=2, bg=BACKGROUND_COLOR, relief="solid", 
                   command=lambda x=x, y=y: handle_click(x, y))
         for y in range(GRID_SIZE)] for x in range(GRID_SIZE)]
    for x, row in enumerate(buttons_computer):
        for y, btn in enumerate(row):
            btn.grid(row=x, column=y)
 
    # Панель состояния
    turns_label = tk.Label(root, text=f"Ходов осталось: {turns_left}", font=("Arial", 12))
    turns_label.grid(row=2, column=0, padx=10, pady=10)
    status_label = tk.Label(root, text="Расставьте свои корабли.", font=("Arial", 12))
    status_label.grid(row=2, column=1, padx=10, pady=10)
 
    root.mainloop()
 
if __name__ == "__main__":
    start_game()

...
    if player_ships == 0:
        status_label.config(text="Компьютер победил!")
        disable_computer_buttons()
...