hnc-daniil/HNC/Exercises/Ship_Battle/main.py

import json
import os
import time
from tkinter import filedialog
from tkinter import *
from ShipField import ShipField

my_field = ShipField()
enemy_field = ShipField()

active_field = my_field


def draw_field(window, field, col_offset):
    buttons = []
    for r in range(0, field.field_size):
        for c in range(0, field.field_size):
            btn = Button(window, text='', width=5, height=2)
            btn.grid(column=c + col_offset, row=r)
            btn.bind('<Button-1>', lambda e, x=r, y=c: left_button_click(buttons, x, y))
            btn.bind('<Button-3>', right_button_click)
            btn.bind('<Enter>', lambda e, x=r, y=c: button_enter(buttons, x, y))
            buttons.append(btn)
    colorize(field, buttons)
    return buttons


def colorize(field, buttons):
    for i in range(len(field.field)):
        bg = "white"
        if field.field[i] == "1":
            bg = 'pink'
        if field.field[i] == "\\":
            bg = 'grey'
        if field.field[i] == "0":
            bg = 'black'
        if field.field[i] == "p":
            bg = 'blue'
        if field.field[i] == "r":
            bg = 'red'
        if "+" in field.field[i]:
            bg = 'orange'
<<<<<<< HEAD
=======

>>>>>>> d51d6b9fd1009b5d2eec1276597e4b6e34e005b5
        buttons[i].configure(bg=bg)


def keypress_handler(e):
    global active_field
    if e.keysym.isnumeric():
        active_field.set_ship_size(int(e.keysym))
    else:
        if e.keysym == 'm':
            active_field.toggle_field_mode()


def left_button_click(buttons, row, col):
    global active_field

    active_field.action(row, col)
    colorize(active_field, buttons)


def right_button_click(d):
    global active_field
    active_field.toggle_ship_direction()


def button_enter(buttons, row, col):
    global active_field

    if buttons == my_buttons:
        active_field = my_field
        enemy_field.clear_marker()
        my_field.target(row, col)

    elif buttons == enemy_buttons:
        active_field = enemy_field
        my_field.clear_marker()
        enemy_field.target(row, col)

    colorize(my_field, my_buttons)
    colorize(enemy_field, enemy_buttons)


def savebutton_click():
    file_path = filedialog.asksaveasfilename(filetypes=[("JSON files", "*.json")])

    if file_path:
        with open(file_path, 'w') as f:
            json.dump({'my_field': my_field}, f, default=ShipField.convert_to_json)


def loadbutton_click():
    global my_field

    file_path = filedialog.askopenfilename(filetypes=[("JSON files", "*.json")])    

    if os.path.isfile(file_path):
        with open(file_path) as lines:
            my_field.from_json(json.load(lines)["my_field"])

    colorize(my_field, my_buttons)


def savebutton_click_enemy():
    file_path = filedialog.asksaveasfilename(filetypes=[("JSON files", "*.json")])

    if file_path:
        with open(file_path, 'w') as f:
            json.dump({'enemy_field': enemy_field}, f, default=ShipField.convert_to_json)


def loadbutton_click_enemy():
    global enemy_field

    file_path = filedialog.askopenfilename(filetypes=[("JSON files", "*.json")])    

    if os.path.isfile(file_path):
        with open(file_path) as lines:
            enemy_field.from_json(json.load(lines)["enemy_field"])

    colorize(enemy_field, enemy_buttons)


window = Tk()
window.title("Ship Craft!")
window.geometry('940x510')
window.bind_all('<KeyPress>', keypress_handler)

my_buttons = draw_field(window, my_field, 0)
enemy_buttons = draw_field(window, enemy_field, 11)

lbl = Label(window, text='', width=5, height=2)
lbl.grid(column=10, row=0)

savebutton = Button(window, text='Save', width=20, height=2, command=savebutton_click)
savebutton.grid(column=0, row=11, columnspan=4)

loadbutton = Button(window, text='Load', width=20, height=2, command=loadbutton_click)
loadbutton.grid(column=5, row=11, columnspan=4)

<<<<<<< HEAD
savebutton_enemy = Button(window, text='Save_enemy', width=20, height=2, command=savebutton_click_enemy)
savebutton_enemy.grid(column=11, row=11, columnspan=4)

loadbutton_enemy = Button(window, text='Load_enemy', width=20, height=2, command=loadbutton_click_enemy)
loadbutton_enemy.grid(column=16, row=11, columnspan=4)


window.mainloop()
=======
window.mainloop()
>>>>>>> d51d6b9fd1009b5d2eec1276597e4b6e34e005b5