diff --git a/app/cinema/__init__.py b/app/cinema/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/app/cinema/bar.py b/app/cinema/bar.py new file mode 100644 index 000000000..23b0421bb --- /dev/null +++ b/app/cinema/bar.py @@ -0,0 +1,8 @@ +from app.people.customer import Customer + + +class CinemaBar: + + @staticmethod + def sell_product(product: str, customer: Customer) -> None: + print(f"Cinema bar sold {product} to {customer.name}.") diff --git a/app/cinema/hall.py b/app/cinema/hall.py new file mode 100644 index 000000000..173668304 --- /dev/null +++ b/app/cinema/hall.py @@ -0,0 +1,19 @@ +from app.people.cinema_staff import Cleaner +from app.people.customer import Customer + + +class CinemaHall: + + def __init__(self, number: int) -> None: + self.number = number + + def movie_session( + self, movie_name: str, + customers: list[Customer], + cleaning_staff: Cleaner + ) -> None: + print(f'"{movie_name}" started in hall number {self.number}.') + for customer in customers: + customer.watch_movie(movie_name) + print(f'"{movie_name}" ended.') + cleaning_staff.clean_hall(self.number) diff --git a/app/main.py b/app/main.py index 7eeef6542..a6aa1b209 100644 --- a/app/main.py +++ b/app/main.py @@ -1,6 +1,23 @@ -# write your imports here +from app.cinema.bar import CinemaBar +from app.cinema.hall import CinemaHall +from app.people.customer import Customer +from app.people.cinema_staff import Cleaner -def cinema_visit(customers: list, hall_number: int, cleaner: str, movie: str): - # write you code here - pass +def cinema_visit( + customers: list[dict], + hall_number: int, + cleaner: str, + movie: str) -> None: + customer_list = [ + Customer(customer["name"], + customer["food"]) for customer in customers] + + cinema_hall = CinemaHall(hall_number) + cinema_bar = CinemaBar() + cleaner = Cleaner(cleaner) + + for customer in customer_list: + cinema_bar.sell_product(customer.food, customer) + + cinema_hall.movie_session(movie, customer_list, cleaner) diff --git a/app/people/__init__.py b/app/people/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/app/people/cinema_staff.py b/app/people/cinema_staff.py new file mode 100644 index 000000000..8317af679 --- /dev/null +++ b/app/people/cinema_staff.py @@ -0,0 +1,7 @@ +class Cleaner: + def __init__(self, name: str) -> None: + self.name = name + + def clean_hall(self, + hall_number: int) -> None: + print(f"Cleaner {self.name} is cleaning hall number {hall_number}.") diff --git a/app/people/customer.py b/app/people/customer.py new file mode 100644 index 000000000..73eba8466 --- /dev/null +++ b/app/people/customer.py @@ -0,0 +1,7 @@ +class Customer: + def __init__(self, name: str, food: str) -> None: + self.name = name + self.food = food + + def watch_movie(self, movie: str) -> None: + print(f'{self.name} is watching "{movie}".')