Создайте систему для управления заказами в ресторане, используя классы для моделирования блюд и заказов

class Dish:
    def __init__(self, name, price):
        self.name = name
        self.price = price
 
class Order:
    def __init__(self, table):
        self.dishes = []
        self.table = table
 
    def add_dish(self, dish):
        self.dishes.append(dish)
 
    def remove_dish(self, dish):
        if dish in self.dishes:
            self.dishes.remove(dish)
        else:
            print("Этого блюда нет в заказе.")
 
    def calculate_total(self):
        total = 0
        for dish in self.dishes:
            total += dish.price
        return total
 
def print_order(order):
    print("Заказ для", order.table)
    for dish in order.dishes:
        print("Блюдо:", dish.name, "| Цена Блюда:", dish.price)
    print("Всего:", order.calculate_total())
 
class MainDish(Dish):
    def __init__(self, name, price_MainDish):
        super().__init__(name, price_MainDish)
        self.price_Dish = price_MainDish
        self.name = name
 
class Dessert(Order):
    def __init__(self, name, prise_Dessert):
        super().__init__(name)
        self.name = name
        self.price = prise_Dessert
 
class Drink(Order):
    def __init__(self, name, price_Drink):
        super().__init__(name)
        self.name = name
        self.price = price_Drink
# Создание блюд
dish1 = Dish("Стейк", 25.99)
dish2 = Dish("Салат", 12.99)
dish3 = Dish("Паста", 18.99)
 
 
#Основные блюда
MainDish1 = MainDish("Картошка", 10.99)
 
#Десерты
Dessert1 = Dessert("Торт", 49.99)
 
# напитки
Drink1= Drink("Лимонад",5.99)
 
#Меню
print('МЕНЮ')
print('Еда')
print(dish1.name, dish1.price)
print(dish2.name, dish2.price)
print(dish3.name, dish3.price)
 
print('Основные блюда')
print(MainDish1.name, MainDish1.price)
 
print('напитки')
print(print_order.__init__())
 
# Создание заказов
# order1 = Order("Столик 1")
# order2 = Order("Столик 2")
#
# # Добавление блюд в заказы
# order1.add_dish(dish1)
# order1.add_dish(dish3)
# order2.add_dish(dish2)
#
# # Вывод информации о заказах
# print_order(order1)
# print_order(order2)
 
while True:
    choice= int(input("""
    1.Mеню
    2.Добавить в заказ
    3.Удалить товар из заказа
    4.Показать заказ
    5.Выйти
    """))
    if choice==1:
        for i in range (len(Dish)):
            print(f'{i+1}.{Dish[i]}')
 
    elif choice==2:
        item = int(input('Выберете товар:'))
        found = False
        try:
            for p in Dish:
                if d==Dish[item-1]:
                    cart.append(products[item-1])
                    print("Товар добавлен")
                    found = True
            if not found:
                raise  IndexError
 
 
        except IndexError:
            print("Товар не найден")
    elif choice==3:
        item = int(input('Выберете товар для удаления:'))
        try:
            if products[item-1] in cart:
                cart.remove(products[item-1])
                print(f'{products[item-1]}(товар удалён)')
            else:
                raise IndexError
        except IndexError:
            print('Товар в корзине не найден')
    elif choice==4:
        if len(cart)>0:
            for i in range (len(cart)):
                print(f'{i+1}.{cart[i]}')
        else:
            print("Корзина пустая!")
    elif choice == 5:
        print('Выход')
        break
    else:
        print('Введите от 1 до 5')

from typing import List, Union
from enum import Enum
 
class DishType(Enum):
    MAIN = "Основное блюдо"
    DESSERT = "Десерт"
    DRINK = "Напиток"
 
class Dish:
    """Базовый класс для блюд"""
    
    def __init__(self, name: str, price: float, dish_type: DishType):
        self.name = name
        self.price = price
        self.dish_type = dish_type
    
    def __str__(self):
        return f"{self.name} - {self.price}₽ ({self.dish_type.value})"
    
    def __repr__(self):
        return f"Dish('{self.name}', {self.price}, {self.dish_type})"
 
class MainDish(Dish):
    """Класс для основных блюд"""
    
    def __init__(self, name: str, price: float, is_vegetarian: bool = False, cooking_time: int = 20):
        super().__init__(name, price, DishType.MAIN)
        self.is_vegetarian = is_vegetarian
        self.cooking_time = cooking_time  # время приготовления в минутах
    
    def __str__(self):
        veg_status = " (Вегетарианское)" if self.is_vegetarian else ""
        return f"{self.name} - {self.price}₽ - {self.cooking_time} мин{veg_status}"
    
    def __repr__(self):
        return f"MainDish('{self.name}', {self.price}, {self.is_vegetarian}, {self.cooking_time})"
 
class Dessert(Dish):
    """Класс для десертов"""
    
    def __init__(self, name: str, price: float, contains_nuts: bool = False, is_gluten_free: bool = False):
        super().__init__(name, price, DishType.DESSERT)
        self.contains_nuts = contains_nuts
        self.is_gluten_free = is_gluten_free
    
    def __str__(self):
        nuts_status = " (С орехами)" if self.contains_nuts else ""
        gluten_status = " (Без глютена)" if self.is_gluten_free else ""
        return f"{self.name} - {self.price}₽{nuts_status}{gluten_status}"
    
    def __repr__(self):
        return f"Dessert('{self.name}', {self.price}, {self.contains_nuts}, {self.is_gluten_free})"
 
class Drink(Dish):
    """Класс для напитков"""
    
    def __init__(self, name: str, price: float, is_alcoholic: bool = False, volume: float = 0.5):
        super().__init__(name, price, DishType.DRINK)
        self.is_alcoholic = is_alcoholic
        self.volume = volume  # объем в литрах
    
    def __str__(self):
        alcohol_status = " (Алкогольный)" if self.is_alcoholic else " (Безалкогольный)"
        return f"{self.name} - {self.price}₽ - {self.volume}л{alcohol_status}"
    
    def __repr__(self):
        return f"Drink('{self.name}', {self.price}, {self.is_alcoholic}, {self.volume})"
 
class Order:
    """Класс для управления заказами"""
    
    def __init__(self, order_id: int = None):
        self.order_id = order_id
        self.dishes: List[Dish] = []
    
    def add_dish(self, dish: Dish) -> None:
        """Добавить блюдо в заказ"""
        self.dishes.append(dish)
        print(f"Добавлено: {dish}")
    
    def remove_dish(self, dish_name: str) -> bool:
        """Удалить блюдо из заказа по названию"""
        for i, dish in enumerate(self.dishes):
            if dish.name.lower() == dish_name.lower():
                removed_dish = self.dishes.pop(i)
                print(f"Удалено: {removed_dish}")
                return True
        print(f"Блюдо '{dish_name}' не найдено в заказе")
        return False
    
    def get_total_price(self) -> float:
        """Получить общую стоимость заказа"""
        return sum(dish.price for dish in self.dishes)
    
    def get_dishes_by_type(self, dish_type: DishType) -> List[Dish]:
        """Получить блюда определенного типа"""
        return [dish for dish in self.dishes if dish.dish_type == dish_type]
    
    def display_order(self) -> None:
        """Отобразить информацию о заказе"""
        if not self.dishes:
            print("Заказ пуст")
            return
        
        print(f"\n=== Заказ #{self.order_id if self.order_id else 'N/A'} ===")
        for i, dish in enumerate(self.dishes, 1):
            print(f"{i}. {dish}")
        
        print(f"\nИтого: {self.get_total_price()}₽")
        print("=" * 40)
    
    # Перегрузка операторов
    def __add__(self, other: 'Order') -> 'Order':
        """Объединение двух заказов"""
        combined_order = Order()
        combined_order.dishes = self.dishes + other.dishes
        return combined_order
    
    def __gt__(self, other: 'Order') -> bool:
        """Сравнение заказов по стоимости"""
        return self.get_total_price() > other.get_total_price()
    
    def __lt__(self, other: 'Order') -> bool:
        """Сравнение заказов по стоимости"""
        return self.get_total_price() < other.get_total_price()
    
    def __eq__(self, other: 'Order') -> bool:
        """Проверка на равенство заказов"""
        return self.get_total_price() == other.get_total_price()
    
    def __len__(self):
        """Количество блюд в заказе"""
        return len(self.dishes)
    
    def __contains__(self, dish_name: str):
        """Проверка наличия блюда в заказе"""
        return any(dish.name.lower() == dish_name.lower() for dish in self.dishes)
 
# Демонстрация работы системы
def main():
    # Создаем блюда
    steak = MainDish("Стейк Рибай", 1200, is_vegetarian=False, cooking_time=25)
    salad = MainDish("Греческий салат", 450, is_vegetarian=True, cooking_time=10)
    cake = Dessert("Тирамису", 350, contains_nuts=False, is_gluten_free=False)
    ice_cream = Dessert("Мороженое", 200, contains_nuts=True, is_gluten_free=True)
    wine = Drink("Красное вино", 800, is_alcoholic=True, volume=0.75)
    juice = Drink("Апельсиновый сок", 250, is_alcoholic=False, volume=0.3)
    
    # Создаем заказы
    order1 = Order(1)
    order2 = Order(2)
    
    print("=== Демонстрация системы управления заказами ===\n")
    
    # Заполняем первый заказ
    print("Формируем заказ #1:")
    order1.add_dish(steak)
    order1.add_dish(wine)
    order1.add_dish(cake)
    order1.display_order()
    
    # Заполняем второй заказ
    print("\nФормируем заказ #2:")
    order2.add_dish(salad)
    order2.add_dish(juice)
    order2.add_dish(ice_cream)
    order2.display_order()
    
    # Демонстрация удаления блюда
    print("\nУдаляем мороженое из заказа #2:")
    order2.remove_dish("Мороженое")
    order2.display_order()
    
    # Демонстрация объединения заказов
    print("\nОбъединяем заказы #1 и #2:")
    combined_order = order1 + order2
    combined_order.display_order()
    
    # Демонстрация сравнения заказов
    print("\nСравнение заказов:")
    print(f"Заказ #1 > Заказ #2: {order1 > order2}")
    print(f"Заказ #1 < Заказ #2: {order1 < order2}")
    print(f"Заказ #1 == Заказ #2: {order1 == order2}")
    
    # Демонстрация фильтрации по типу
    print(f"\nОсновные блюда в объединенном заказе:")
    main_dishes = combined_order.get_dishes_by_type(DishType.MAIN)
    for dish in main_dishes:
        print(f"  - {dish}")
    
    # Демонстрация проверки наличия блюда
    print(f"\nПроверка наличия блюд:")
    print(f"Стейк в заказе #1: {'Стейк Рибай' in order1}")
    print(f"Пицца в заказе #1: {'Пицца' in order1}")
    
    # Демонстрация количества блюд
    print(f"\nКоличество блюд в заказах:")
    print(f"Заказ #1: {len(order1)} блюд")
    print(f"Заказ #2: {len(order2)} блюд")
    print(f"Объединенный заказ: {len(combined_order)} блюд")
 
if __name__ == "__main__":
    main()