How do I create a custom context manager in Python?

How do I create a custom context manager in Python?

You can create a custom context manager by defining a class with __enter__ and __exit__ methods.

Can you show me a simple example?

Sure! Here's a basic example:

class MyContext:
    def __enter__(self):
        print('Entering the context')
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        print('Exiting the context')

with MyContext() as context:
    print('Inside the context')

What happens if an exception occurs inside the context?

If an exception occurs, it will be passed to the __exit__ method, where you can handle it.

Can I use a generator to create a context manager?

Yes! You can use the contextlib module's contextmanager decorator to create a context manager with a generator.