diff --git a/src/entities/Player.py b/src/entities/Player.py index c47fc35..a7aba49 100644 --- a/src/entities/Player.py +++ b/src/entities/Player.py @@ -10,9 +10,11 @@ class Player(Entity): super().__init__("player.jpg", size, (spawnpoint[0] * size, spawnpoint[1] * size)) # Where the player is facing, 0 - north, 1 self.rotation = Rotations.NORTH + self.statistics = Statistics(100, 0, 0, 100) # Move in a desired direction def move(self, rotation): + self.statistics.set_stamina(-1) if rotation.value == Rotations.NORTH.value: self.rect.y -= self.rect.w elif rotation.value == Rotations.EAST.value: @@ -22,14 +24,34 @@ class Player(Entity): elif rotation.value == Rotations.WEST.value: self.rect.x -= self.rect.w + # Returns given statistic + def getStatistic(self, stat): + if stat.value == StatisticNames.HP: + return self.statistics.hp + elif stat.value == StatisticNames.HUNGER: + return self.statistics.hunger + elif stat.value == StatisticNames.THIRST: + return self.statistics.thirst + elif stat.value == StatisticNames.STAMINA: + return self.statistics.stamina + return None + def rotate(self, rotation): # If the player is not facing given direction, it will not move the first time, it will only get rotated if self.rotation.value != rotation.value: self.image = pygame.transform.rotate(self.image, ((self.rotation.value - rotation.value) * 90)) self.rotation = rotation + class Rotations(Enum): NORTH = 0 EAST = 1 SOUTH = 2 - WEST = 3 \ No newline at end of file + WEST = 3 + + +class StatisticNames(Enum): + HP = 0 + STAMINA = 1 + HUNGER = 2 + THIRST = 3 \ No newline at end of file