Làm thế nào để sử dụng Python đóng quản lý bối cảnh

Question

Các tiêu chuẩn thư viện open chức năng vừa là một chức năng:

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

hoặc như một người quản lý bối cảnh:

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

Tôi cố gắng để bắt chước hành vi này bằng contextlib.closing , trong đó File là tệp tùy chỉnh I/O của tôi:

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

này làm việc như mong đợi như một người quản lý bối cảnh:

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

nhưng tất nhiên nếu tôi gọi trực tiếp, tôi lấy lại đối tượng closing thay vì đối tượng của tôi:

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

Vì vậy, làm thế nào để thực hiện myopen để cả hai hoạt động như một trình quản lý ngữ cảnh và trả về đối tượng Tệp của tôi khi được gọi trực tiếp?

dụ làm việc đầy đủ trên github: https://gist.github.com/1352573

Answer 1

Điều đơn giản nhất có lẽ là để thực hiện các __enter__ và __exit__ phương pháp cho mình. Một cái gì đó như thế này nên làm điều đó:

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

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

Nó sẽ, bằng cách này, có nhiều thành ngữ để làm công việc của phương pháp open của bạn trong phương pháp __init__ của bạn.