CODEWITHSUNDEEP
pythoncontextmanager
caring-goat-913
caring-goat-913

Reputation: 4049

How can I access attributes of a context manager?

I want to accomplish something like:

class my_context(object):
    def __init__(self):
        self.obj1 = Obj()
        self.obj2 = Obj()
        ...

    def __enter__(self):
        ''' initialize objects '''

    def __exit__(self, type, value, tb):
        ''' uninitialize objects '''

There are many Obj attributes which are resources which need to be be closed/deleted/etc. I was hoping to use a context manager to set them up and then get rid of them. However what I've found is that I cannot access the attributes when I try:

with my_context() as cont:
    cont.obj1  # doesn't work

Is there a way I can access these attributes?

Upvotes: 2

Views: 1141

Answers (1)

Kevin
Kevin

Reputation: 30151

For the with... as syntax to work, your __enter__() must return a value.

If you want to use the same attributes as your my_context class provides, you probably want to return self.

Upvotes: 6

Related Questions