diff --git a/app/main.py b/app/main.py index 8fbf3053..a0afd8fc 100644 --- a/app/main.py +++ b/app/main.py @@ -1 +1,44 @@ -# write your code here +class Animal: + def __init__(self, name: str, + appetite: int, + is_hungry: bool = True) -> None: + self.name = name + self.appetite = appetite + self.is_hungry = is_hungry + + def print_name(self) -> str: + print(f"Hello, I\'m {self.name}") + + def feed(self) -> str | int: + if self.is_hungry: + print(f"Eating {self.appetite} food points...") + self.is_hungry = False + return self.appetite + else: + return 0 + + +class Cat(Animal): + def __init__(self, name: str, is_hungry: bool = True) -> None: + super().__init__(name, appetite=3, is_hungry=is_hungry) + + def catch_mouse(self) -> str: + print("The hunt began!") + + +class Dog(Animal): + def __init__(self, name: str, is_hungry: bool = True) -> None: + super().__init__(name, appetite=7, is_hungry=is_hungry) + + def bring_slippers(self) -> str: + print("The slippers delivered!") + + +def feed_animals(list_of_animals: list[Animal]) -> int: + food = 0 + for animal in list_of_animals: + if animal.is_hungry: + animal.feed() + food += animal.appetite + print(food) + return food