python/282073.html">上下文管理器最常用的是確保正確關閉文件，
with open('/path/to/file', 'r') as f: f.read()with 語句的基本語法，
with expression [as variable]:
with-block
expression是一個上下文管理器，其實現了enter和exit兩個函數。當我們調用一個with語句時， 依次執行一下步驟，
1.首先生成一個上下文管理器expression， 比如open('xx.txt').
2.執行expression.enter()。如果指定了[as variable]說明符，將enter()的返回值賦給variable.
3.執行with-block語句塊.
4.執行expression.exit(),在exit()函數中可以進行資源清理工作.
with語句不僅可以管理文件，還可以管理鎖、連接等等，如，
#管理鎖import threadinglock = threading.lock()with lock: #執行一些操作 pass#數據庫連接管理def test_write(): sql = """ #具體的sql語句 """ con = DBConnection() with con as cursor: cursor.execute(sql) cursor.execute(sql) cursor.execute(sql)
自定義上下文管理器
上下文管理器就是實現了上下文協議的類 ,也就是要實現 __enter__()和__exit__()兩個方法。
__enter__()：主要執行一些環境準備工作，同時返回一資源對象。如果上下文管理器open("test.txt")的enter()函數返回一個文件對象。
__exit__()：完整形式為exit(type, value, traceback),這三個參數和調用sys.exec_info()函數返回值是一樣的，分別為異常類型、異常信息和堆棧。如果執行體語句沒有引發異常，則這三個參數均被設為None。否則，它們將包含上下文的異常信息。_exit()方法返回True或False,分別指示被引發的異常有沒有被處理，如果返回False，引發的異常將會被傳遞出上下文。如果exit()函數內部引發了異常，則會覆蓋掉執行體的中引發的異常。處理異常時，不需要重新拋出異常，只需要返回False，with語句會檢測exit()返回False來處理異常。
class test_query: def __init__(self): pass def query(self): print('query') def __enter__(self): # 如果是數據庫連接就可以返回cursor對象 # cursor = self.cursor # return cursor print('begin query,') return self #由于這里沒有資源對象就返回對象 def __exit__(self,exc_type,exc_value,traceback): if traceback is None: print('End') else: print('Error') # 如果是數據庫連接提交操作 # if traceback is None: # self.commit() # else: # self.rollback() if __name__ == '__main__': with test_query() as q: q.query()contextlib
編寫enter和exit仍然很繁瑣，因此Python的標準庫contextlib提供了更簡單的寫法，
基本語法
@contextmanager def some_generator(<arguments>): <setup> try: yield <value> finally: <cleanup>
生成器函數some_generator就和我們普通的函數一樣，它的原理如下：
some_generator函數在在yield之前的代碼等同于上下文管理器中的enter函數.
yield的返回值等同于enter函數的返回值，即如果with語句聲明了as <variable>，則yield的值會賦給variable.
然后執行<cleanup>代碼塊，等同于上下文管理器的exit函數。此時發生的任何異常都會再次通過yield函數返回。
例，
自動加括號
import contextlib@contextlib.contextmanagerdef tag(name): print('<{}>'.format(name)) yield print('</{}>'.format(name))if __name__ == '__main__': with tag('h1') as t: print('hello') print('context')管理鎖
@contextmanagerdef locked(lock): lock.acquire() try: yield finally: lock.release()with locked(myLock): #代碼執行到這里時，myLock已經自動上鎖 pass #執行完后會，會自動釋放鎖
管理文件關閉
@contextmanagerdef myopen(filename, mode="r"): f = open(filename,mode) try: yield f finally: f.close()with myopen("test.txt") as f: for line in f: print(line)管理數據庫回滾
@contextmanagerdef transaction(db): db.begin() try： yield except: db.rollback() raise else: db.commit()with transaction(mydb): mydb.cursor.execute(sql) mydb.cursor.execute(sql) mydb.cursor.execute(sql) mydb.cursor.execute(sql)
這就很方便！
如果一個對象沒有實現上下文，我們就不能把它用于with語句。這個時候，可以用closing()來把該對象變為上下文對象。它的exit函數僅僅調用傳入參數的close函數.
例如，用with語句使用urlopen()：
import contextlibfrom urllib.request import urlopenif __name__ == '__main__': with contextlib.closing(urlopen('https://www.python.org')) as page: for line in page: print(line)closing也是一個經過@contextmanager裝飾的generator，這個generator編寫起來其實非常簡單：@contextmanagerdef closing(thing): try: yield thing finally: thing.close()嵌套使用
import contextlibfrom urllib.request import urlopenif __name__ == '__main__': with contextlib.closing(urlopen('https://www.python.org')) as page,/ contextlib.closing(urlopen('https://www.python.org')) as p: for line in page: print(line) print(p)在2.x中需要使用contextlib.nested()才能使用嵌套，3.x中可以直接使用。
更多函數可以參考官方文檔https://docs.python.org/3/library/contextlib.html
新聞熱點
疑難解答
