# Context Managers上下文管理器#用來有效的管理資源，之前在講文件讀寫操作中#有提到過#現在以文件讀寫為例#演示 Context Managers上下文管理器#普通的文件寫入操作：f = open('test.txt','w') f.write('寫入數據到文件！')f.close()#接下來使用 Context Managers的方式完成：with open('test_new.txt','w') as f:  f.write('寫入數據到文件！')#這里我們并沒有去寫關閉文件的操作#因為當我們完成自己的代碼操作，退出#with上下文管理器后，文件會自動關閉#上下文管理器常用在數據庫的連接關閉#還可以獲取釋放鎖等操作
#接下來我們自定義一個上下文管理器類#用來打開文件class Open_File():  #初始化文件名和文件打開的模式  def __init__(self,filename,mode):    self.filename = filename    self.mode = mode  #設置上下文管理器的實際操作  #打開文件    def __enter__(self):    self.file = open(self.filename,self.mode)    return self.file  #退出管理器時，執行關閉文件操作  def __exit__(self,exc_type,exc_val,traceback):    self.file.close()#使用上下文管理器#with這行代碼塊實際執行了#Open_File類中的 __init__#__enter__兩個方法with Open_File('sample.txt','w') as f:  f.write('testing')#我們打印文件的狀態print(f.closed)#True
#接下來使用裝飾器來重寫上面的#自定義上下文管理器#需要導入模塊contexlibfrom contextlib import contextmanager
@contextmanagerdef open_file(file,mode):  try:    f = open(file,mode)    yield f  finally:    f.close()#調用管理器with open_file('sample_new.txt','w') as wf:  wf.write('寫入數據！')print(wf.closed)
#接下來看一個os模塊中的操作import os
#獲取當前工作目錄cwd = os.getcwd()#切換到指定目錄test1os.chdir('test1')#列出該目錄中的所有內容print(os.listdir())#切換回當前工作目錄os.chdir(cwd)
#獲取當前工作目錄cwd = os.getcwd()#切換到指定目錄test2os.chdir('test2')#列出該目錄中的所有內容print(os.listdir())#切換回當前工作目錄os.chdir(cwd)#我們要做同樣的事情很多次#運行我們得到結果如下：#['test1.doc', 'test1.txt']#['test2.doc', 'test2.txt']#上邊的代碼邏輯非常適合#使用上下文管理器實現#修改如下：@contextmanagerdef change_dir(destination):  try:    cwd = os.getcwd()    os.chdir(destination)    #這里yield返回所有    yield  finally:    os.chdir(cwd)#調用：with change_dir('test1'):  print(os.listdir())with change_dir('test2'):  print(os.listdir())