diff --git a/app/main.py b/app/main.py index fa56336e..cfbb909a 100644 --- a/app/main.py +++ b/app/main.py @@ -1 +1,46 @@ -# write your code here +from __future__ import annotations + + +class Animal: + + alive = [] + + def __init__( + self, + name: str, + health: int = 100, + hidden: bool = False + ) -> None: + self.name = name + self.health = health + self.hidden = hidden + Animal.alive.append(self) + + def __str__(self) -> str: + return (f"{{" + f"Name: {self.name}, " + f"Health: {self.health}, " + f"Hidden: {self.hidden}}}") + + def __repr__(self) -> str: + return self.__str__() + + +class Herbivore(Animal): + + def hide(self) -> None: + if self.hidden is False: + self.hidden = True + else: + self.hidden = False + + +class Carnivore(Animal): + + def bite(self, animal: Herbivore) -> None: + if animal.hidden is True or isinstance(animal, Carnivore): + return + else: + animal.health -= 50 + if animal.health <= 0: + Animal.alive.remove(animal)