def get_damage(self, damage):
"""Take damage points and change model color.
The more damage the unit taken the more red it'll be.
Args:
damage (int): Damage points to get.
"""
Unit.get_damage(self, damage)
self.model.setColorScale(self.model.getColorScale()[0] + 0.018, 1, 1, 1)
def get_damage(self, damage):
"""Getting damage.
Args:
damage (int): Damage points to get.
"""
if base.labels.TRAITS[2][0] in self.traits: # noqa: F821
# Masochism
self.energy += 1
Unit.get_damage(self, damage)