How do I implement the Composite Pattern in Python?

How do I implement the Composite Pattern in Python?

The Composite Pattern allows you to treat individual objects and compositions of objects uniformly. Would you like a code example?

Yes, please provide a code example.

Here's a simple example:

class Component:
    def operation(self):
        pass

class Leaf(Component):
    def operation(self):
        return "Leaf"

class Composite(Component):
    def __init__(self):
        self.children = []

    def add(self, component):
        self.children.append(component)

    def operation(self):
        results = [child.operation() for child in self.children]
        return f'Composite: [{

You can create instances of Leaf and Composite, add leaves to the composite, and call the operation method:

leaf1 = Leaf()
leaf2 = Leaf()
composite = Composite()
composite.add(leaf1)
composite.add(leaf2)
print(composite.operation())  # Output: Composite: [Leaf, Leaf]