What are some advanced examples of using contextlib in Python?

What are some advanced examples of using contextlib in Python?

Contextlib is a module in Python that provides utilities for working with context managers. Some advanced examples include creating your own context managers using contextlib.contextmanager, and using contextlib.ExitStack to manage multiple context managers dynamically.

Can you show me an example of a custom context manager?

Sure! Here's a simple example:

from contextlib import contextmanager

@contextmanager
def my_context():
    print('Entering the context')
    yield
    print('Exiting the context')

with my_context():
    print('Inside the context')

What does ExitStack do?

ExitStack allows you to manage multiple context managers and ensures that they are all exited properly, even if an error occurs. It's useful when you have a variable number of context managers.

Can you give an example of using ExitStack?

Certainly! Here's an example:

from contextlib import ExitStack

with ExitStack() as stack:
    f1 = stack.enter_context(open('file1.txt'))
    f2 = stack.enter_context(open('file2.txt'))
    # Do something with f1 and f2