diff --git a/src/entities/Player.py b/src/entities/Player.py index d9a3c0c..a7aba49 100644 --- a/src/entities/Player.py +++ b/src/entities/Player.py @@ -24,6 +24,18 @@ 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: @@ -35,4 +47,11 @@ 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