diff --git a/app/main.py b/app/main.py index fa56336e..ab69f29a 100644 --- a/app/main.py +++ b/app/main.py @@ -1 +1,38 @@ -# write your code here +class Animal: + alive = [] + + def __init__(self, + name: str, + health: int = 100, + hidden: bool = False + ) -> None: + self.health = health + self.name = name + self.hidden = hidden + Animal.alive.append(self) + + def die(self) -> None: + if self.health <= 0 and self in Animal.alive: + Animal.alive.remove(self) + + def __repr__(self) -> str: + return "{{Name: {}, Health: {}, Hidden: {}}}".format( + self.name, self.health, self.hidden + ) + + @classmethod + def __str__(cls) -> str: + return str(cls.alive) + + +class Herbivore(Animal): + def hide(self) -> None: + self.hidden = not self.hidden + + +class Carnivore(Animal): + def bite(self, animal: Herbivore) -> None: + if isinstance(animal, Herbivore) and not animal.hidden: + animal.health -= 50 + if animal.health <= 0: + animal.die()