Cómo utilizar Python cierre gestor de contexto

Question

La función estándar de la biblioteca open funciona tanto como una función:

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

o como un gestor de contexto:

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

Estoy tratando de imitar este comportamiento usando contextlib.closing , donde File es mi clase personalizada de E/S de archivo:

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

esto funciona como se esperaba como un gestor de contexto:

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

pero por supuesto si llamo directamente, regrese al closing objeto en lugar de mi objetivo:

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

Así que, ¿Cómo implemento myopen para que funcione como administrador de contexto y devuelve mi objeto File cuando se llama directamente?

ejemplo de trabajo completo en github: https://gist.github.com/1352573

Answer 1

Lo más fácil es probablemente para implementar los métodos __enter__ y __exit__ mismo. Algo así debe hacerlo:

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

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

Sería, por cierto, ser más idiomática para hacer el trabajo de su método de open en su método __init__.