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..2396fec05 --- /dev/null +++ b/app/cinema/hall.py @@ -0,0 +1,17 @@ +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..0071a5b69 100644 --- a/app/main.py +++ b/app/main.py @@ -1,6 +1,22 @@ -# write your imports here +from app.cinema.bar import CinemaBar +from app.cinema.hall import CinemaHall +from app.people.cinema_staff import Cleaner +from app.people.customer import Customer -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_name: str, movie: str) -> None: + + hall = CinemaHall(hall_number) + bar_func = CinemaBar() + cleaning = Cleaner(cleaner_name) + list_of_customer = [] + + for customer_info in customers: + customer = Customer(customer_info["name"], customer_info["food"]) + list_of_customer.append(customer) + bar_func.sell_product(product=customer_info["food"], customer=customer) + + CinemaHall.movie_session(self=hall, movie_name=movie, + customers=list_of_customer, + cleaning_staff=cleaning) 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..ca2b5a90c --- /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..00b3da32c --- /dev/null +++ b/app/people/customer.py @@ -0,0 +1,8 @@ +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}".')