如何使用Python关闭上下文管理器

Question

标准库open函数既可以作为函数使用:

f = open('file.txt')
print(type(f))
<type 'file'>

或作为上下文管理者:

with open('file.txt') as f:
    print(type(f))
<type 'file'>

我试图模仿这种行为contextlib.closing,File我的自定义文件I/O类在哪里:

def myopen(filename):
    f = File(filename)
    f.open()
    return closing(f)

这可以像上下文管理器一样工作:

with myopen('file.txt') as f:
    print(type(f))
<class '__main__.File'>

但是当然如果我直接打电话,我会收回closing对象而不是我的对象:

f = myopen(filename)
print(type(f))
<class 'contextlib.closing'>

那么,我myopen该如何实现它以便它既作为上下文管理器又直接调用时返回我的File对象？

github上的完整工作示例:https: //gist.github.com/1352573

Answer 1

最简单的事情可能是自己实现__enter__和__exit__方法.这样的事情应该这样做:

class File(object):
   # ... all the methods you already have ...

   # context management
   def __enter__(self):
       return self
   def __exit__(self, *exc_info):
       self.close()

顺便说一句,在您的open方法中执行__init__方法的工作会更加惯用.