上下文管理器最常用的是確保正確關(guān)閉文件晓猛，
with open('/path/to/file', 'r') as f:
f.read()
with 語(yǔ)句的基本語(yǔ)法戒职，
with expression [as variable]:
with-block
expression是一個(gè)上下文管理器透乾，其實(shí)現(xiàn)了enter和exit兩個(gè)函數(shù)磕秤。當(dāng)我們調(diào)用一個(gè)with語(yǔ)句時(shí)， 依次執(zhí)行一下步驟市咆，
1.首先生成一個(gè)上下文管理器expression， 比如open('xx.txt').
2.執(zhí)行expression.enter()蒙兰。如果指定了[as variable]說(shuō)明符，將enter()的返回值賦給variable.
3.執(zhí)行with-block語(yǔ)句塊.
4.執(zhí)行expression.exit(),在exit()函數(shù)中可以進(jìn)行資源清理工作.
with語(yǔ)句不僅可以管理文件搜变，還可以管理鎖、連接等等挠他，如，
#管理鎖
import threading
lock = threading.lock()
with lock:
#執(zhí)行一些操作
pass
#數(shù)據(jù)庫(kù)連接管理
def test_write():
sql = """ #具體的sql語(yǔ)句
"""
con = DBConnection()
with con as cursor:
cursor.execute(sql)
cursor.execute(sql)
cursor.execute(sql)
自定義上下文管理器
上下文管理器就是實(shí)現(xiàn)了上下文協(xié)議的類(lèi) ,也就是要實(shí)現(xiàn) __enter__()和__exit__()兩個(gè)方法殖侵。
__enter__()：主要執(zhí)行一些環(huán)境準(zhǔn)備工作，同時(shí)返回一資源對(duì)象愉耙。如果上下文管理器open("test.txt")的enter()函數(shù)返回一個(gè)文件對(duì)象贮尉。
__exit__()：完整形式為exit(type, value, traceback),這三個(gè)參數(shù)和調(diào)用sys.exec_info()函數(shù)返回值是一樣的，分別為異常類(lèi)型败砂、異常信息和堆棧。如果執(zhí)行體語(yǔ)句沒(méi)有引發(fā)異常昌犹，則這三個(gè)參數(shù)均被設(shè)為None。否則斜姥，它們將包含上下文的異常信息。_exit()方法返回True或False,分別指示被引發(fā)的異常有沒(méi)有被處理铸敏，如果返回False，引發(fā)的異常將會(huì)被傳遞出上下文杈笔。如果exit()函數(shù)內(nèi)部引發(fā)了異常，則會(huì)覆蓋掉執(zhí)行體的中引發(fā)的異常蒙具。處理異常時(shí)球榆，不需要重新拋出異常持钉，只需要返回False，with語(yǔ)句會(huì)檢測(cè)exit()返回False來(lái)處理異常融师。
class test_query:
def __init__(self):
pass
def query(self):
print('query')
def __enter__(self):
# 如果是數(shù)據(jù)庫(kù)連接就可以返回cursor對(duì)象
# cursor = self.cursor
# return cursor
print('begin query,')
return self #由于這里沒(méi)有資源對(duì)象就返回對(duì)象
def __exit__(self,exc_type,exc_value,traceback):
if traceback is None:
print('End')
else:
print('Error')
# 如果是數(shù)據(jù)庫(kù)連接提交操作
# if traceback is None:
# self.commit()
# else:
# self.rollback()
if __name__ == '__main__':
with test_query() as q:
q.query()
contextlib
編寫(xiě)enter和exit仍然很繁瑣，因此Python的標(biāo)準(zhǔn)庫(kù)contextlib提供了更簡(jiǎn)單的寫(xiě)法舀射，
基本語(yǔ)法
@contextmanager
def some_generator(<arguments>):
<setup>
try:
yield <value>
finally:
<cleanup>
生成器函數(shù)some_generator就和我們普通的函數(shù)一樣，它的原理如下：
some_generator函數(shù)在在yield之前的代碼等同于上下文管理器中的enter函數(shù).
yield的返回值等同于enter函數(shù)的返回值脆烟，即如果with語(yǔ)句聲明了as <variable>，則yield的值會(huì)賦給variable.
然后執(zhí)行<cleanup>代碼塊邢羔，等同于上下文管理器的exit函數(shù)。此時(shí)發(fā)生的任何異常都會(huì)再次通過(guò)yield函數(shù)返回拜鹤。
例，
自動(dòng)加括號(hào)
import contextlib
@contextlib.contextmanager
def tag(name):
print('<{}>'.format(name))
yield
print('</{}>'.format(name))
if __name__ == '__main__':
with tag('h1') as t:
print('hello')
print('context')
管理鎖
@contextmanager
def locked(lock):
lock.acquire()
try:
yield
finally:
lock.release()
with locked(myLock):
#代碼執(zhí)行到這里時(shí)敏簿，myLock已經(jīng)自動(dòng)上鎖
pass
#執(zhí)行完后會(huì)，會(huì)自動(dòng)釋放鎖
管理文件關(guān)閉
@contextmanager
def 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)
管理數(shù)據(jù)庫(kù)回滾
@contextmanager
def 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)
這就很方便惯裕！
如果一個(gè)對(duì)象沒(méi)有實(shí)現(xiàn)上下文绣硝，我們就不能把它用于with語(yǔ)句。這個(gè)時(shí)候鹉胖，可以用closing()來(lái)把該對(duì)象變?yōu)樯舷挛膶?duì)象。它的exit函數(shù)僅僅調(diào)用傳入?yún)?shù)的close函數(shù).
例如甫菠，用with語(yǔ)句使用urlopen()：
import contextlib
from urllib.request import urlopen
if __name__ == '__main__':
with contextlib.closing(urlopen('https://www.python.org')) as page:
for line in page:
print(line)
closing也是一個(gè)經(jīng)過(guò)@contextmanager裝飾的generator，這個(gè)generator編寫(xiě)起來(lái)其實(shí)非常簡(jiǎn)單：
@contextmanager
def closing(thing):
try:
yield thing
finally:
thing.close()
嵌套使用
import contextlib
from urllib.request import urlopen
if __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中可以直接使用。
更多函數(shù)可以參考官方文檔https://docs.python.org/3/library/contextlib.html
