import pygame
from models.Object import Object
from models.Waiter import Waiter


class Kitchen(Object):
    def __init__(self, square_size, screen_size, left_square, top_square):
        super().__init__(
            'kitchen',
            square_size,
            screen_size,
            left_square,
            top_square
        )

    def take_orders(self, waiter: Waiter):
        for table in waiter.get_order_list():
            table.done_order()