Come utilizzare Python chiusura contesto direttore

Question

La funzione di libreria standard open funziona sia come una funzione:

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

o come un contesto allenatore:

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

Sto cercando di imitare questo comportamento utilizzando contextlib.closing , dove File è il mio file personalizzato classe I/O:

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

questo funziona come previsto da un contesto allenatore:

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

ma ovviamente se chiamo direttamente, torno l'oggetto invece del mio oggetto closing:

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

Quindi, come si implementa myopen in modo che funzioni entrambi come gestore di contesto e restituisce il mio oggetto File quando viene chiamato direttamente?

esempio funzionante completa su github: https://gist.github.com/1352573

Answer 1

La cosa più semplice è probabilmente quello di attuare le __enter__ e __exit__ metodi da soli. Qualcosa del genere dovrebbe farlo:

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

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

Sarebbe, tra l'altro, essere più idiomatico per fare il lavoro del metodo open nel metodo __init__.