Added getStatistic method & enum

This commit is contained in:
Marcin Kostrzewski 2020-04-05 22:04:40 +02:00
parent dc7023adfa
commit 5c80033a0b

View File

@ -24,6 +24,18 @@ class Player(Entity):
elif rotation.value == Rotations.WEST.value: elif rotation.value == Rotations.WEST.value:
self.rect.x -= self.rect.w 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): 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 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: if self.rotation.value != rotation.value:
@ -35,4 +47,11 @@ class Rotations(Enum):
NORTH = 0 NORTH = 0
EAST = 1 EAST = 1
SOUTH = 2 SOUTH = 2
WEST = 3 WEST = 3
class StatisticNames(Enum):
HP = 0
STAMINA = 1
HUNGER = 2
THIRST = 3