您好,登錄后才能下訂單哦!
這篇文章主要介紹“python內(nèi)置模塊之怎么實現(xiàn)上下文管理contextlib”的相關(guān)知識,小編通過實際案例向大家展示操作過程,操作方法簡單快捷,實用性強,希望這篇“python內(nèi)置模塊之怎么實現(xiàn)上下文管理contextlib”文章能幫助大家解決問題。
Python中當我們們打開文本時,通常會是用with語句,with語句允許我們非常方便的使用資源,而不必擔心資源沒有關(guān)閉。
with open('/path/filename', 'r') as f: f.read()
然而,并不是只有open()函數(shù)返回fp對象才能使用 with 語句。實際上,任何對象,只要正確實現(xiàn)上下文管理,就可以使用with語句。
實現(xiàn)上下文管理是通過 __enter__ 和 __exit__ 這兩個方法實現(xiàn)的。例如,下面的class實現(xiàn)了這兩個方法:
class Query(object): def __init__(self, name): self.name = name def __enter__(self): print('Begin') return self def __exit__(self, exc_type, exc_value, traceback): if exc_type: print('Error') else: print('End') def query(self): print('Query info about %s...' % self.name)
這樣我們可以把自己寫的資源對象用于 with 語句。
with Query('Bob') as q: q.query()
編寫 __enter__ 和 __exit__ 仍然很繁瑣,因此Python的標準庫 contextlib 提供了更簡單的寫法,上面的代碼可以改寫為:
from contextlib import contextmanager class Query(object): def __init__(self, name): self.name = name def query(self): print('Query info about %s...' % self.name) @contextmanager def create_query(name): print('Begin') q = Query(name) yield q print('End')
@contextmanager 這個裝飾器接受一個 generator,用 yield 語句把 with ... as var 把變量輸出去,然后,with 語句就可以正常的工作了:
with create_query('Bob') as q: q.query()
很多時候,我們希望在某段代碼執(zhí)行前后自動執(zhí)行特定代碼,也可以用 @contextmanager實現(xiàn)。
@contextmanager def tag(name): print("<%s>" % name) yield print("" % name) with tag("h2"): print("hello") print("world")
上述代碼執(zhí)行結(jié)果:
hello
world
</h2>
代碼的執(zhí)行順序是:
with 語句 首先執(zhí)行 yield 之前的語句,因此打印出.
yield 調(diào)用會執(zhí)行 with 語句內(nèi)部的所有語句,因此打印出 hello 和 world.
最后執(zhí)行yield之后的語句,打印出.
如果一個對象沒有實現(xiàn)上下文,就不能使用 with 語句,但是可以用 closing() 來把對象變?yōu)樯舷挛膶ο蟆?/p>
from contextlib import closing from urllib.request import urlopen with closing(urlopen('https://www.python.org')) as page: for line in page: print(line)
closing 也是一個經(jīng)過 @contextmanager 裝飾的generator
@contextmanager def closing(thing): try: yield thing finally: thing.close()
它的作用就是把任意對象變?yōu)樯舷挛膶ο螅⒅С?with語句。
關(guān)于“python內(nèi)置模塊之怎么實現(xiàn)上下文管理contextlib”的內(nèi)容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識,可以關(guān)注億速云行業(yè)資訊頻道,小編每天都會為大家更新不同的知識點。
免責聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。