diff --git a/app/main.py b/app/main.py index 8fbf3053..378e3e16 100644 --- a/app/main.py +++ b/app/main.py @@ -1 +1,42 @@ -# 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) -> None: + print(f"Hello, I'm {self.name}") + + def feed(self) -> int: + if self.is_hungry: + print(f"Eating {self.appetite} food points...") + return self.appetite + else: + return 0 + + +class Cat(Animal): + def __init__(self, name: str, is_hungry: bool = True) -> None: + super().__init__(appetite=3) + self.name = name + self.is_hungry = is_hungry + + def catch_mouse(self) -> None: + print("The hunt began!") + + +class Dog(Animal): + def __init__(self, name: str, is_hungry: bool = True) -> None: + super().__init__(appetite=7) + self.name = name + self.is_hungry = is_hungry + + def bring_slippers(self) -> None: + print("The slippers delivered!") + + +def feed_animals(animals: list[Animal]) -> int: + sum = 0 + for item in animals: + sum += item.appetite + return sum