import pygame class Enemy: def __init__(self, screen): self.pos = [0,0] self.size = 20 self.health = 200 self.speed = 3 self.screen = screen self.color = (255,0,0) def target(self, obj): self.target = obj def draw(self): pygame.draw.rect(self.screen, self.color, pygame.rect.Rect(self.pos, (self.size, self.size))) def move(self, x, y): self.pos[0] += x * self.speed self.pos[1] += y * self.speed def update(self): dx = self.target.pos[0] - self.pos[0] dy = self.target.pos[1] - self.pos[1] length = (dx**2+dy**2)**(0.5) x = dx/length y = dy/length self.move(x,y) self.check_collisions() def check_collisions(self): for bullet in self.target.bullets: if self.pos[0] <= bullet.pos[0] <= self.pos[0]+self.size and self.pos[1] <= bullet.pos[1] <= self.pos[1]+self.size: self.health -= 20 self.target.bullets.remove(bullet)