# open 方法的返回值賦值給變量 f，當離開 with 代碼塊的時候，系統(tǒng)會自動調(diào)用 f.close() 方法
# with 的作用和使用 try/finally 語句是一樣的。
with open("output.txt", "r") as f:
  f.write("XXXXX")

# 自定義的MyFile類
# 實現(xiàn)了 __enter__() 和 __exit__() 方法的對象都可稱之為上下文管理器
class MyFile():
  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
  # with代碼塊執(zhí)行完或者with中發(fā)生異常，就會自動執(zhí)行__exit__方法。
  def __exit__(self, *args):
    print("will exit")
    self.f.close()
# 會自動調(diào)用MyFile對象的__enter__方法,并將返回值賦給f變量。
with MyFile('out.txt', 'w') as f:
  print("writing")
  f.write('hello, python')
  # 當with代碼塊執(zhí)行結(jié)束,或出現(xiàn)異常時,會自動調(diào)用MyFile對象的__exit__方法。
 

from contextlib import contextmanager
@contextmanager
def my_open(path, mode):
  f = open(path, mode)
  yield f
  f.close()
# 將my_open函數(shù)中yield后的變量值賦給f變量。
with my_open('out.txt', 'w') as f:
  f.write("XXXXX")
  # 當with代碼塊執(zhí)行結(jié)束,或出現(xiàn)異常時,會自動執(zhí)行yield后的代碼。