~~~ f = open("test.txt", "w") try: f.write("aaa") except IOError: print("error") finally: f.close() ~~~ ~~~ with open("test.txt", "r") as f: f.write("aaa") ~~~ 以上兩端代碼效果一樣 with 的作用和使用 try/finally 語句是一樣的。那么它的實現原理是什么？ ### 上下文管理器 ``` 任何實現了 __enter__() 和 __exit__() 方法的對象都可稱之為上下文管理器，上下文管理器對象可以使用 with 關鍵 字。顯然，文件（file）對象也實現了上下文管理器 ``` ~~~ class File(): def __init__(self, filename, mode): self.filename = filename self.mode = mode def __enter__(self): print("entering") self.f = open(self.filename, self.mode) return self.f def __exit__(self, *args): print("will exit") self.f.close() ~~~ Python 還提供了一個 contextmanager 的裝飾器，更進一步簡化了上下文管理器的實現方式。通過 yield 將函數分割成兩部分，yield 之前的語句在 \_\_enter\_\_ 方法中執行，yield 之后的語句在 \_\_exit\_\_ 方法中執行。緊跟在 yield 后面的值是函數的返回值。 ~~~ from contextlib import contextmanager @contextmanager def my_open(path, mode): f = open(path, mode) yield f f.close() ~~~