python中如何使用Context Managers上下文管理器

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