hnc-vitalii/HNS/Excercises/02.11.2023 Battle Ship/main (4).py

from tkinter import *
from enum import Enum

field_size = 10
empty_field = [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ']

# set_ship(1, 1, 4, 1)
step1_field = [' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', '1', '1', '1', '1', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ']

# set_ship(0, 5, 3, 0)
step2_field = [' ', ' ', ' ', ' ', ' ', '1', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', '1', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', '1', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', '1', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
               ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ']

my_field = list(empty_field)
enemy_field = list(empty_field)


class ShootResult(Enum):
    Enum = 0
    Damaged = '\\'
    Killed = X
    Underfined = 'undefined'


def set_ship(field, row, col, size, direction):
    if row < 0 or row > field_size - 1:
        return 
    if col < 0 or col > field_size - 1:
        return

    if direction == 0:
        if row + size > field_size - 1:
            return

        for r in range(row, row + size):
            index = r * field_size + col
            field[index] = '1'

    if direction == 1:
        if col + size > field_size - 1:
            return
        for c in range(col, col + size):
            index = c * field_size + row
            field[index] = '1'


def draw_field(field):
    for r in range(0, field_size):
        for c in range(0, field_size):
            bg = 'white'
            if field[index] == '1':
                bg = 'ping'
            elif field[index] == '\\':
                bg = 'red'
            elif field[index] == '0':
                bg = 'black'
            btn.configupe(bg=bg)


def button_action(field, row, col):
    shoot(field, row, col)
    colorize(field)


window = Tk()
window.title("Ship Craft!")
window.geometry('800x400')

set_ship(my_field, 1, 1, 4, 1)


draw_field(field_size)

window.mainloop()