enemy

 1import pygame.draw
 2"""pygame.draw used to more easily render enemies"""
 3
 4
 5class Enemy:
 6    """Class that handles the creation of enemies"""
 7    def __init__(self, e_type, health, speed, strength, path):
 8        self._type = e_type
 9        """The type of enemy, to be used at a later date for different kinds of enemies"""
10        self._health = health
11        self._max_health = health
12        self._speed = speed
13        self._strength = strength
14        self._path = path
15        """The path the enemy is supposed to follow"""
16        self._position = self._path[0]
17        """Initially set at the starting point on the path"""
18        self._path_index = 0
19        self._status = True
20        """Means the enemy is alive"""
21        self._resource_worth = health * 3
22
23    def _move(self):
24        """Function to move the enemy"""
25        if self._path_index < len(self._path) - 1:
26            """Checks to see if the enemy has anywhere else to go."""
27            target_x, target_y = self._path[self._path_index]
28            """Finds the target x and y"""
29            dir_x, dir_y = target_x - self._position[0], target_y - self._position[1]
30            """Sets the direction to go"""
31            distance = (dir_x ** 2 + dir_y ** 2) ** 0.5
32            """Helps adjust distance and move based on speed stat"""
33            if distance < self._speed:
34                self._position = (target_x, target_y)
35                self._path_index += 1
36                """Move to the next point"""
37            else:
38                self._position = (
39                    self._position[0] + dir_x / distance * self._speed,
40                    self._position[1] + dir_y / distance * self._speed,
41                )
42
43    def take_damage(self, damage):
44        """Takes damage and calls kill_enemy if health hits zero"""
45        self._health = self._health - damage
46        if self.get_enemy_health() <= 0:
47            self.kill_enemy()
48
49    def kill_enemy(self):
50        """Kills the enemy and calls reward_resources"""
51        self._status = False
52
53    def reward_resources(self, base):
54        """Gives player resources based on how much the enemy is worth"""
55        base.add_money(self._resource_worth)
56
57    def damage_base(self, base):
58        """Damages the base and removes the enemy"""
59        base.remove_health(self.get_enemy_strength())
60        self._status = False
61
62    def is_alive(self):
63        """Returns the enemy status"""
64        return self._status
65
66    def get_enemy_type(self):
67        """Returns the enemy type"""
68        return self._type
69
70    def get_enemy_health(self):
71        """Returns the enemy's health"""
72        return self._health
73
74    def get_enemy_speed(self):
75        """Returns the enemy's speed"""
76        return self._speed
77
78    def get_enemy_strength(self):
79        """Returns the enemy's strength"""
80        return self._strength
81
82    def render(self, window):
83        """Renders the enemy, currently as a circle. Plan to render as different shapes based on enemy type"""
84        pygame.draw.circle(window, (255, 0, 0),
85                           (int(self._position[0]), int(self._position[1])), 10)
86        self._draw_health_bar(window)
87
88    def _draw_health_bar(self, window):
89        """Creates a visual for the player to know the status of the enemy's health"""
90        health_ratio = self._health / self._max_health
91        """Finds ratio of health"""
92        green = int(255 * health_ratio)
93        red = 255 - green
94        pygame.draw.rect(window, (red, green, 0),
95                         (self._position[0] - 15, self._position[1] - 20, 30 * health_ratio, 5))