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..44f7b8408 --- /dev/null +++ b/app/cinema/bar.py @@ -0,0 +1,7 @@ +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..3890b740a --- /dev/null +++ b/app/cinema/hall.py @@ -0,0 +1,14 @@ +from app.people.cinema_staff import Cleaner + + +class CinemaHall: + def __init__(self, number: int) -> None: + self.number = number + + def movie_session(self, movie_name: str, customers: list, + cleaning_staff: "Cleaner") -> None: + print(f'"{movie_name}" started in hall number {self.number}.') + for customer in customers: + customer.watch_movie(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..bb669d835 100644 --- a/app/main.py +++ b/app/main.py @@ -1,6 +1,19 @@ -# 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, hall_number: int, + cleaner: str, movie: str) -> None: + customer_objects = [] + + for customer_info in customers: + customer = Customer(name=customer_info["name"], + food=customer_info["food"]) + customer_objects.append(customer) + CinemaBar.sell_product(product=customer.food, customer=customer) + + hall = CinemaHall(hall_number) + hall.movie_session(movie_name=movie, customers=customer_objects, + cleaning_staff=Cleaner(name=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..059cb0dcb --- /dev/null +++ b/app/people/cinema_staff.py @@ -0,0 +1,8 @@ +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..1fe92ec11 --- /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}".')