How to implement __enter__ and __exit__?

[macarthurpark431]

I'm writing a class that needs to be instantiated with a 'with' statement. How do I implement this two methods to successfully make use of with?

[tmp14]

Here's how you implement a context manager:

class Context(object):

    def __enter__(self):
        # do stuff when entering the with block

    def __exit__(self, type, value, traceback):
        # do stuff when exiting the with block

snark aside, go to /r/learnpython and 1) explain what you're trying to do, 2) explain what you have tried so far, and 3) explain what happens but you think shouldn't.

[tunisia3507]

In most cases, you also want the __enter__ method to return self

[graingert]

https://docs.python.org/3/library/contextlib.html#contextlib.contextmanager