python中的自动售货机-分类和总价

class VendingMachine:
    def __init__(self, category):
        self.category = category
        self.products = {
            '饮料': {'矿泉水': 2.5, '果汁': 3.0, '碳酸饮料': 2.0},
            '零食': {'薯片': 5.0, '巧克力': 4.0, '饼干': 3.5},
            '咖啡': {'美式咖啡': 6.0, '拿铁': 7.0, '卡布奇诺': 8.0},
            '快餐': {'汉堡': 10.0, '炸鸡': 12.0, '薯条': 6.0}
        }
        self.selected_products = {}
        self.total_price = 0.0

    def select_product(self, product, quantity):
        if product in self.products[self.category]:
            self.selected_products[product] = quantity
            self.total_price += self.products[self.category][product] * quantity
        else:
            print("该商品不在售货机的分类中。")

    def calculate_total_price(self):
        return self.total_price

# 示例用法
vm = VendingMachine('饮料')
vm.select_product('矿泉水', 2)
vm.select_product('碳酸饮料', 1)
total_price = vm.calculate_total_price()
print("购买商品的总价为：", total_price)