getitem-1-9.py

class Item:
    def __init__(self, name, qty=0):
        self.name = name
        self.qty = qty

    def __str__(self):
        return f"{self.name} - {self.qty} left"

class VendingMachine:
    def __init__(self):
        self.items_on_sale = []

    def __getitem__(self, idx):
        if isinstance(idx, int):
            return self.items_on_sale[idx]
        raise TypeError("Index should be an integer")

    def add_items(self, items):
        """
        Add items to the vending machine
        :param items: sequence of Item objects
        """
        self.items_on_sale.extend(items)

    def buy_item(self, idx):
        item = self[idx]
        if item.qty > 0:
            item.qty -= 1
            return item.name
        print("This slot is empty")
        return None  # Explicitly returning None, for clarity


hideous_machine = VendingMachine()
hideous_machine.add_items(
    [
        Item("Hazelnut Chocolate", 5),
        Item("Dark Chocolate", 3),
        Item("Still Water", 7),
        Item("Sparkling Water", 4),
    ]
)

print(hideous_machine[2])
hideous_machine.buy_item(2)
print(hideous_machine[2])