Шутер ООП

import tkinter as tk
import random
import math
from random import randint

# Константы
WIDTH, HEIGHT = 800, 600
PLAYER_SIZE = 20
ENEMY_SIZE = 20
BULLET_SIZE = 5
PLAYER_SPEED = 3
ENEMY_SPEED = 4
BULLET_SPEED = 10


class Game:
    def __init__(self):
        self.root = tk.Tk()
        self.root.title("Стрелялка")
        self.canvas = tk.Canvas(self.root, width=WIDTH, height=HEIGHT, bg="black")
        self.canvas.pack()

        self.player = Player(self)
        self.enemies: list[Enemy] = []
        self.bullets = []
        self.score = 0

        self.root.bind("<B3-Motion>", self.player.move)
        self.root.bind("<B1-Motion>", self.shoot)
        # self.root.bind("<1>", self.shoot)

        self.spawn_enemy()
        self.game_loop()
        self.root.mainloop()

    def shoot(self, event):
        bullet = Bullet(self, event.x, event.y)
        self.bullets.append(bullet)

    def spawn_enemy(self):
        boss = True if randint(0, 5) == 0 else False
        if boss:
            speed = ENEMY_SPEED / 2
            color = "green"
            enemy = Enemy(self, color, speed, 40)
            enemy.hp = 5
        else:
            enemy = Enemy(self, None, None, None)
        self.enemies.append(enemy)
        self.root.after(1240, self.spawn_enemy)

    def update_bullets(self):
        for bullet in self.bullets[:]:
            if not bullet.update():
                self.bullets.remove(bullet)

    def update_enemies(self):
        for enemy in self.enemies[:]:
            if not enemy.update():
                try:
                    self.enemies.remove(enemy)
                except:
                    pass

    def game_loop(self):
        self.update_bullets()
        self.update_enemies()
        self.root.title(f"Счёт: {self.score}")
        self.root.after(40, self.game_loop)

    def game_over(self):
        self.canvas.create_text(WIDTH // 2, HEIGHT // 2, text="GAME OVER", font=("Arial", 50), fill="white")
        self.root.after_cancel(self.game_loop)


class Player:
    def __init__(self, game):
        self.game = game
        self.body = game.canvas.create_oval(
            WIDTH // 2 - PLAYER_SIZE, HEIGHT // 2 - PLAYER_SIZE,
            WIDTH // 2 + PLAYER_SIZE, HEIGHT // 2 + PLAYER_SIZE,
            fill="blue"
        )

    def move(self, event):
        self.game.canvas.coords(
            self.body,
            event.x - PLAYER_SIZE, event.y - PLAYER_SIZE,
            event.x + PLAYER_SIZE, event.y + PLAYER_SIZE
        )


class Bullet:
    def __init__(self, game, target_x, target_y):
        self.game = game
        x, y, _, _ = game.canvas.coords(game.player.body)
        self.body = game.canvas.create_oval(
            x - BULLET_SIZE, y - BULLET_SIZE,
            x + BULLET_SIZE, y + BULLET_SIZE,
            fill="yellow"
        )
        self.angle = math.atan2(target_y - y, target_x - x)

    def update(self):
        self.game.canvas.move(
            self.body, BULLET_SPEED * math.cos(self.angle), BULLET_SPEED * math.sin(self.angle)
        )
        x, y, _, _ = self.game.canvas.coords(self.body)
        if x < 0 or x > WIDTH or y < 0 or y > HEIGHT:
            self.game.canvas.delete(self.body)
            return False
        return True


class Enemy:
    def __init__(self, game, color, speed, size):
        self.game = game
        self.speed = speed if speed else ENEMY_SPEED
        self.color = color if color else 'red'
        self.hp = 2
        self.size = size if size else ENEMY_SIZE
        side = random.choice(["top", "bottom", "left", "right"])
        if side == "top":
            self.x, self.y = random.randint(0, WIDTH), 0
        elif side == "bottom":
            self.x, self.y = random.randint(0, WIDTH), HEIGHT
        elif side == "left":
            self.x, self.y = 0, random.randint(0, HEIGHT)
        else:
            self.x, self.y = WIDTH, random.randint(0, HEIGHT)

        self.body = game.canvas.create_rectangle(
            self.x - self.size, self.y - self.size,
            self.x + self.size, self.y + self.size,
            fill=self.color
        )

    def update(self):
        px, py, _, _ = self.game.canvas.coords(self.game.player.body)
        ex, ey, _, _ = self.game.canvas.coords(self.body)
        angle = math.atan2(py - ey, px - ex)
        self.game.canvas.move(self.body, self.speed * math.cos(angle), self.speed * math.sin(angle))

        # Проверка столкновения с игроком
        if (px - ex) ** 2 + (py - ey) ** 2 < (PLAYER_SIZE + self.size) ** 2:
            self.game.game_over()
            return False

        # Проверка столкновения с пулями
        for bullet in self.game.bullets[:]:
            bx, by, _, _ = self.game.canvas.coords(bullet.body)
            if (bx - ex) ** 2 + (by - ey) ** 2 < (BULLET_SIZE + self.size) ** 2:
                if self.hp > 1:
                    self.hp -= 1
                    self.game.canvas.delete(bullet.body)
                    self.game.bullets.remove(bullet)
                else:
                    self.game.canvas.delete(self.body)
                    self.game.enemies.remove(self)
                    self.game.canvas.delete(bullet.body)
                    self.game.bullets.remove(bullet)
                    self.game.score += 1
                    return False
        return True


if __name__ == "__main__":
    Game()