diff --git a/HNC/Exercises/Ship_Battle/ShipDirection.py b/HNC/Exercises/Ship_Battle/ShipDirection.py new file mode 100644 index 0000000..72f4354 --- /dev/null +++ b/HNC/Exercises/Ship_Battle/ShipDirection.py @@ -0,0 +1,6 @@ +from enum import Enum + +class ShipDirection(Enum): + VERTICAL = "VERTICAL" + HORIZONTAL = "HORIZONTAL" + UNKNOWN = "UNKNOWN" \ No newline at end of file diff --git a/HNC/Exercises/Ship_Battle/ShipField.py b/HNC/Exercises/Ship_Battle/ShipField.py index a87c565..50a9e9a 100644 --- a/HNC/Exercises/Ship_Battle/ShipField.py +++ b/HNC/Exercises/Ship_Battle/ShipField.py @@ -1,4 +1,7 @@ +import copy from ShootResult import ShootResult +from ShipMode import ShipMode +from ShipDirection import ShipDirection class ShipField: @@ -17,9 +20,9 @@ class ShipField: self.ships = [4, 3, 3, 2, 2, 2, 1, 1, 1, 1] self.field_size = 10 - self.field_mode = 0 + self.field_mode = ShipMode.PUT self.ship_size = 4 - self.ship_direction = 0 + self.ship_direction = ShipDirection.VERTICAL def __getitem__(self, item): @@ -38,27 +41,30 @@ class ShipField: def action(self, row, col): self.clear_marker() - if self.field_mode == 0: - if self.ship_size in self.ships and self.check_possible(row, col): + if self.field_mode == ShipMode.PUT: + if self.check_ship(row, col): + self.get_ship(row, col) + + elif self.ship_size in self.ships and self.check_possible(row, col): self.set_ship(row, col) - elif self.field_mode == 1: + elif self.field_mode == ShipMode.SHOOT: self.shoot(row, col) def target(self, row, col): self.clear_marker() - if self.field_mode == 0: + if self.field_mode == ShipMode.PUT: if self.check_possible(row, col): - if self.ship_direction == 0: + if self.ship_direction == ShipDirection.VERTICAL: for r in range(row, row + self.ship_size): if self.ship_size in self.ships: self.field[r * self.field_size + col] = "p" else: self.field[r * self.field_size + col] = "r" - if self.ship_direction == 1: + if self.ship_direction == ShipDirection.HORIZONTAL: for c in range(col, col + self.ship_size): if self.ship_size in self.ships: self.field[row * self.field_size + c] = "p" @@ -83,13 +89,13 @@ class ShipField: if col < 0 or col > self.field_size: return index = row * self.field_size + col - if self.ship_direction == 0: + if self.ship_direction == ShipDirection.VERTICAL: if self.field_size - row < self.ship_size: return for r in range(row, row + self.ship_size): index = r * self.field_size + col self.field[index] = "1" - if self.ship_direction == 1: + if self.ship_direction == ShipDirection.HORIZONTAL: if self.field_size - col < self.ship_size: return for c in range(col, col + self.ship_size): @@ -99,6 +105,59 @@ class ShipField: if self.ship_size in self.ships: self.ships.remove(self.ship_size) + def get_ship(self, row, col): + if row < 0 or row > self.field_size: + return + if col < 0 or col > self.field_size: + return + + self.field[row * self.field_size + col] = '' + + ship_size = 1 + ship_direction = ShipDirection.UNKNOWN + + + # check vertical + for r in range(row + 1, self.field_size): + if self.check_ship(r, col): + ship_size += 1 + ship_direction = ShipDirection.VERTICAL + self.field[r * self.field_size + col] = '' + else: + break + + for r in range(row - 1, -1, -1): + if self.check_ship(r, col): + ship_size += 1 + ship_direction = ShipDirection.VERTICAL + self.field[r * self.field_size + col] = '' + else: + break + + + if ship_direction == ShipDirection.UNKNOWN: + # check horizontal + for c in range(col + 1, self.field_size): + if self.check_ship(row, c): + ship_size += 1 + ship_direction = ShipDirection.HORIZONTAL + self.field[row * self.field_size + c] = '' + else: + break + + for c in range(col - 1, - 1, -1): + if self.check_ship(row, c): + ship_size += 1 + ship_direction = ShipDirection.HORIZONTAL + self.field[row * self.field_size + c] = '' + else: + break + + self.set_ship_direction(ship_direction) + self.set_ship_size(ship_size) + + self.ships.append(ship_size) + def shoot(self, row, col): if row < 0 or row > self.field_size - 1: return ShootResult.UNDEFINED @@ -113,16 +172,19 @@ class ShipField: return ShootResult.DAMAGED else: return ShootResult.UNDEFINED + + def check_ship(self, row, col): + return self.field[row * self.field_size + col].strip() == '1' def check_possible(self, row, col): - if self.ship_direction == 0: + if self.ship_direction == ShipDirection.VERTICAL: if self.field_size - row >= self.ship_size: for r in range(row, row + self.ship_size): if not self.check_blocked(r, col): return False return True - if self.ship_direction == 1: + if self.ship_direction == ShipDirection.HORIZONTAL: if self.field_size - col >= self.ship_size: for c in range(col, col + self.ship_size): if not self.check_blocked(row, c): @@ -154,24 +216,24 @@ class ShipField: if value is None: return - if type(value) is str and value.isnumeric(): - value = int(value) - - if type(value) is int and 0 <= value <= 1: + if type(value) is not ShipDirection: + return + + if value != ShipDirection.UNKNOWN: self.ship_direction = value def toggle_ship_direction(self): - if self.field_mode == 0: - if self.ship_direction == 0: - self.ship_direction = 1 + if self.field_mode == ShipMode.PUT: + if self.ship_direction == ShipDirection.VERTICAL: + self.ship_direction = ShipDirection.HORIZONTAL else: - self.ship_direction = 0 + self.ship_direction = ShipDirection.VERTICAL def toggle_field_mode(self): - if self.field_mode == 0: - self.field_mode = 1 + if self.field_mode == ShipMode.PUT: + self.field_mode = ShipMode.SHOOT else: - self.field_mode = 0 + self.field_mode = ShipMode.PUT def print_field(self): for r in range(0, self.field_size): @@ -181,4 +243,13 @@ class ShipField: blocked_string += str(self.check_blocked(r, c))[0] + ", " ship_string += self.field[r * self.field_size + c] + ', ' print(blocked_string[:-2] + ' ' + ship_string[:-2]) - print("********************************************************************") \ No newline at end of file + print("********************************************************************") + + + @staticmethod + def convert_to_json(obj): + if isinstance(obj, ShipField): + result = copy.copy(obj.__dict__) + result['field_mode'] = obj.field_mode.value + result['ship_direction'] = obj.ship_direction.value + return result \ No newline at end of file diff --git a/HNC/Exercises/Ship_Battle/ShipField_test.py b/HNC/Exercises/Ship_Battle/ShipField_test.py index bc5ceb7..f12ede3 100644 --- a/HNC/Exercises/Ship_Battle/ShipField_test.py +++ b/HNC/Exercises/Ship_Battle/ShipField_test.py @@ -1,6 +1,8 @@ from unittest import TestCase from ShipField import ShipField from ShootResult import ShootResult +from ShipMode import ShipMode +from ShipDirection import ShipDirection class TestShipField(TestCase): @@ -45,20 +47,16 @@ class TestShipField(TestCase): def test_toggle_field_mode(self): ship_field = ShipField() # Заводим объект типа ShipField - self.assertEqual(ship_field.field_mode, 0) # Проверяем, что изначальное значение field_mode равно 0 + self.assertEqual(ship_field.field_mode, ShipMode.PUT) # Проверяем, что изначальное значение field_mode равно 0 ship_field.toggle_field_mode() # Вызываем метод, который тестируем - self.assertEqual(ship_field.field_mode, 1) # Проверяем, что field_mode принял желаемое значение + self.assertEqual(ship_field.field_mode, ShipMode.SHOOT) # Проверяем, что field_mode принял желаемое значение ship_field.toggle_field_mode() # Вызываем метод, который тестируем - self.assertEqual(ship_field.field_mode, 0) # Проверяем, что field_mode принял желаемое значение + self.assertEqual(ship_field.field_mode, ShipMode.PUT) # Проверяем, что field_mode принял желаемое значение def test_action(self): - ship_field = ShipField() - self.assertEqual(ship_field.action, 0) - - ship_field.action() - self.assertEqual(ship_field.action, 1) + self.fail() def test_target(self): self.fail() @@ -67,10 +65,12 @@ class TestShipField(TestCase): ship_field = ShipField() ship_field.field[0] = 'p' ship_field.field[ship_field.field_size - 1] = 'p' + ship_field.field[ship_field.field_size - 4] = 'r' ship_field.clear_marker() self.assertNotIn('p', ship_field.field) + self.assertNotIn('r', ship_field.field) def test_shoot_empty(self): ship_field = ShipField() @@ -133,7 +133,7 @@ class TestShipField(TestCase): def test_set_ship_size4_horizontal_direction(self): ship_field = ShipField() ship_field.set_ship_size(4) - ship_field.set_ship_direction(1) + ship_field.set_ship_direction(ShipDirection.HORIZONTAL) ship_field.set_ship(6, 3) @@ -154,18 +154,56 @@ class TestShipField(TestCase): self.assertEqual(new_field_string, old_field_string) def test_check_possible(self): - self.fail() + ship_field = ShipField() + ship_field.set_ship_size(4) + ship_field.set_ship_direction(ShipDirection.HORIZONTAL) + + ship_field.set_ship(5, 3) + + self.assertEqual(ship_field.check_possible(5, 3), False) + self.assertEqual(ship_field.check_possible(5, 4), False) + self.assertEqual(ship_field.check_possible(5, 5), False) + self.assertEqual(ship_field.check_possible(5, 6), False) + + self.assertEqual(ship_field.check_possible(6, 3), False) + self.assertEqual(ship_field.check_possible(6, 4), False) + self.assertEqual(ship_field.check_possible(6, 5), False) + self.assertEqual(ship_field.check_possible(6, 6), False) + + self.assertEqual(ship_field.check_possible(7, 3), True) + self.assertEqual(ship_field.check_possible(7, 4), True) + self.assertEqual(ship_field.check_possible(7, 5), True) + self.assertEqual(ship_field.check_possible(7, 6), True) def test_check_blocked(self): - self.fail() + ship_field = ShipField() + ship_field.set_ship_size(4) + ship_field.set_ship_direction(ShipDirection.HORIZONTAL) + + ship_field.set_ship(5, 3) + + self.assertEqual(ship_field.check_possible(5, 3), False) + self.assertEqual(ship_field.check_possible(5, 4), False) + self.assertEqual(ship_field.check_possible(5, 5), False) + self.assertEqual(ship_field.check_possible(5, 6), False) + + self.assertEqual(ship_field.check_possible(6, 3), False) + self.assertEqual(ship_field.check_possible(6, 4), False) + self.assertEqual(ship_field.check_possible(6, 5), False) + self.assertEqual(ship_field.check_possible(6, 6), False) + + self.assertEqual(ship_field.check_possible(7, 3), True) + self.assertEqual(ship_field.check_possible(7, 4), True) + self.assertEqual(ship_field.check_possible(7, 5), True) + self.assertEqual(ship_field.check_possible(7, 6), True) def test_set_ship_direction(self): ship_field = ShipField() - ship_field.set_ship_direction(1) - self.assertEqual(ship_field.ship_direction, 1) + ship_field.set_ship_direction(ShipDirection.HORIZONTAL) + self.assertEqual(ship_field.ship_direction, ShipDirection.HORIZONTAL) - ship_field.set_ship_direction(0) - self.assertEqual(ship_field.ship_direction, 0) + ship_field.set_ship_direction(ShipDirection.VERTICAL) + self.assertEqual(ship_field.ship_direction, ShipDirection.VERTICAL) def test_toggle_ship_direction(self): @@ -173,9 +211,9 @@ class TestShipField(TestCase): ship_field.toggle_ship_direction() - self.assertEqual(ship_field.ship_direction, 1) + self.assertEqual(ship_field.ship_direction, ShipDirection.HORIZONTAL) ship_field.toggle_ship_direction() - self.assertEqual(ship_field.ship_direction, 0) + self.assertEqual(ship_field.ship_direction, ShipDirection.VERTICAL) diff --git a/HNC/Exercises/Ship_Battle/ShipMode.py b/HNC/Exercises/Ship_Battle/ShipMode.py new file mode 100644 index 0000000..5517662 --- /dev/null +++ b/HNC/Exercises/Ship_Battle/ShipMode.py @@ -0,0 +1,5 @@ +from enum import Enum + +class ShipMode(Enum): + PUT = "PUT" + SHOOT = "SHOOT" \ No newline at end of file diff --git a/HNC/Exercises/Ship_Battle/__pycache__/ShipDirection.cpython-311.pyc b/HNC/Exercises/Ship_Battle/__pycache__/ShipDirection.cpython-311.pyc new file mode 100644 index 0000000..d82802b Binary files /dev/null and b/HNC/Exercises/Ship_Battle/__pycache__/ShipDirection.cpython-311.pyc differ diff --git a/HNC/Exercises/Ship_Battle/__pycache__/ShipField.cpython-311.pyc b/HNC/Exercises/Ship_Battle/__pycache__/ShipField.cpython-311.pyc index b442d41..c3e554d 100644 Binary files a/HNC/Exercises/Ship_Battle/__pycache__/ShipField.cpython-311.pyc and b/HNC/Exercises/Ship_Battle/__pycache__/ShipField.cpython-311.pyc differ diff --git a/HNC/Exercises/Ship_Battle/__pycache__/ShipField_test.cpython-311-pytest-8.1.1.pyc b/HNC/Exercises/Ship_Battle/__pycache__/ShipField_test.cpython-311-pytest-8.1.1.pyc index f41d633..add44a6 100644 Binary files a/HNC/Exercises/Ship_Battle/__pycache__/ShipField_test.cpython-311-pytest-8.1.1.pyc and b/HNC/Exercises/Ship_Battle/__pycache__/ShipField_test.cpython-311-pytest-8.1.1.pyc differ diff --git a/HNC/Exercises/Ship_Battle/__pycache__/ShipMode.cpython-311.pyc b/HNC/Exercises/Ship_Battle/__pycache__/ShipMode.cpython-311.pyc new file mode 100644 index 0000000..85cfbec Binary files /dev/null and b/HNC/Exercises/Ship_Battle/__pycache__/ShipMode.cpython-311.pyc differ diff --git a/HNC/Exercises/Ship_Battle/files.py b/HNC/Exercises/Ship_Battle/files.py new file mode 100644 index 0000000..c501b51 --- /dev/null +++ b/HNC/Exercises/Ship_Battle/files.py @@ -0,0 +1,28 @@ +import os + + +def list_levels(): + files = os.listdir(path='.') + result = [] + for file in files: + if file.lower().endswith('.txt'): + result.append(file) + + return result + + +def load(file): + # 1. Загрузить весь файл в строку + f = open(file, 'r') + s = f.read() + + # 2. Заменить все пробелы и переводы строки на "ничего" + s = s.replace(' ', '') + s = s.replace('\n', '') + + # 3. Из полученной строки создать массив, используя разделитель "," + a = s.split(',') + + # a - массив чисел из а + f.close() + return a \ No newline at end of file diff --git a/HNC/Exercises/Ship_Battle/game1.json b/HNC/Exercises/Ship_Battle/game1.json new file mode 100644 index 0000000..badab37 --- /dev/null +++ b/HNC/Exercises/Ship_Battle/game1.json @@ -0,0 +1,7 @@ +{ + "my_field":{ + "field": [], + "ships": [], + "field_mode": "PUT" + } +} \ No newline at end of file diff --git a/HNC/Exercises/Ship_Battle/main.py b/HNC/Exercises/Ship_Battle/main.py index ad1e62c..cbe06d1 100644 --- a/HNC/Exercises/Ship_Battle/main.py +++ b/HNC/Exercises/Ship_Battle/main.py @@ -1,3 +1,4 @@ +import json from tkinter import * from ShipField import ShipField @@ -32,10 +33,10 @@ def colorize(field, buttons): bg = 'black' if field.field[i] == "p": bg = 'blue' - if field.field[i] == "r": - bg = 'red' if "+" in field.field[i]: bg = 'orange' + if "r" in field[i]: + bg = 'red' buttons[i].configure(bg=bg) @@ -76,24 +77,41 @@ def button_enter(buttons, row, col): colorize(my_field, my_buttons) colorize(enemy_field, enemy_buttons) + +def savebutton_click(event): + with open('test.json', 'w') as f: + json.dump({'my_field': my_field}, f, default=ShipField.convert_to_json) + + +def loadbutton_click(event): + data_list = [] + with open('test.json') as lines: + data = json.load(lines) + for i in data['shapes']: + sh1 = create_shape(i) + data_list.append(sh1) + + return data_list + + window = Tk() window.title("Ship Craft!") -window.geometry('940x410') +window.geometry('940x510') window.bind_all('', keypress_handler) -my_field.toggle_ship_direction() -my_field.set_ship_size(4) -my_field.set_ship(1, 1) -my_field.toggle_ship_direction() -my_field.set_ship_size(3) -my_field.set_ship(0, 6) -my_field.set_ship_size(1) -my_field.set_ship(7, 3) - 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) +savebutton.bind('', savebutton_click) +savebutton.grid(column=0, row=11, columnspan=4) + +loadbutton = Button(window, text='Load', width=20, height=2) +loadbutton.bind('', loadbutton_click) +loadbutton.grid(column=5, row=11, columnspan=4) + + window.mainloop() \ No newline at end of file diff --git a/HNC/Exercises/Ship_Battle/test.json b/HNC/Exercises/Ship_Battle/test.json new file mode 100644 index 0000000..3041a88 --- /dev/null +++ b/HNC/Exercises/Ship_Battle/test.json @@ -0,0 +1 @@ +{"shapes": {"field": ["1", "1", "1", "1", " ", " ", " ", " ", "", "1", "", "", " ", " ", " ", "1", " ", "", "", "1", "", "", "", "", "", "", "", "1", "", "", "1", "", "", "", "", "1", "", "", "", "", "1", "", "", "1", "", "", "", "1", " ", " ", "1", "", "", "1", " ", " ", " ", "", "", "", " ", "", "", "", " ", " ", "", "", "", "", " ", "1", "", " ", " ", " ", " ", " ", "", "", " ", "1", "", " ", " ", " ", " ", " ", " ", " ", " ", "1", "", " ", " ", " ", "r", "", "1", "1"], "ships": [], "field_size": 10, "field_mode": "PUT", "ship_size": 1, "ship_direction": "VERTICAL"}} \ No newline at end of file