Need help with "with-statement-compatible" object

[Dmitry Teslenko]

Hello!
I've made some class that can be used with "with statement". It looks this way:

class chdir_to_file( object ):
....
def __enter__(self) :
....

def __exit__(self, type, val, tb):
....
def get_chdir_to_fi le(file_path):
return chdir_to_file(f ile_path)
....

Snippet with object instantiation looks like this:
for s in sys.argv[1:]:
c = chdir_to_file( s )
with c:
print 'Current directory is %s' % os.path.realpat h( os.curdir )

That works fine. I want to enable it to be used in more elegant way:
for s in ... :
with get_chdir_to_fi le( s ) as c:
c.do_something( )

But python complains c is of NoneType and has no "do_something() ". Am
I missing something?

[Peter Otten]

Dmitry Teslenko wrote:

Hello!
I've made some class that can be used with "with statement". It looks this way:

class chdir_to_file( object ):
...
def __enter__(self) :
...

def __exit__(self, type, val, tb):
...
def get_chdir_to_fi le(file_path):
return chdir_to_file(f ile_path)
...

Snippet with object instantiation looks like this:
for s in sys.argv[1:]:
c = chdir_to_file( s )
with c:
print 'Current directory is %s' % os.path.realpat h( os.curdir )

That works fine. I want to enable it to be used in more elegant way:
for s in ... :
with get_chdir_to_fi le( s ) as c:
c.do_something( )

But python complains c is of NoneType and has no "do_something() ". Am
I missing something?

Does the chdir_to_file class have a do_something() method? If so,
changing chdir_to_file._ _enter__() to

def __enter__(self) :
# ...
return self

should help.

Peter

[Dmitry Teslenko]

On Dec 19, 2007 12:14 PM, Peter Otten <__*******@web. dewrote:

def __enter__(self) :
# ...
return self

should help.

That helps. Thanks!