python內(nèi)置模塊之上下文管理contextlib
Python中當(dāng)我們們打開文本時(shí),通常會(huì)是用with語句,with語句允許我們非常方便的使用資源,而不必?fù)?dān)心資源沒有關(guān)閉。
with open('/path/filename', 'r') as f: f.read()
然而,并不是只有open()函數(shù)返回fp對象才能使用 with 語句。實(shí)際上,任何對象,只要正確實(shí)現(xiàn)上下文管理,就可以使用with語句。
實(shí)現(xiàn)上下文管理是通過 __enter__ 和 __exit__ 這兩個(gè)方法實(shí)現(xiàn)的。例如,下面的class實(shí)現(xiàn)了這兩個(gè)方法:
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()
一、@contextmanager
編寫 __enter__ 和 __exit__ 仍然很繁瑣,因此Python的標(biāo)準(zhǔn)庫 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 這個(gè)裝飾器接受一個(gè) generator,用 yield 語句把 with ... as var 把變量輸出去,然后,with 語句就可以正常的工作了:
with create_query('Bob') as q: q.query()
很多時(shí)候,我們希望在某段代碼執(zhí)行前后自動(dòng)執(zhí)行特定代碼,也可以用 @contextmanager實(shí)現(xiàn)。
@contextmanager def tag(name): print("<%s>" % name) yield print("" % name) with tag("h1"): print("hello") print("world")
上述代碼執(zhí)行結(jié)果:
hello
world
</h1>
代碼的執(zhí)行順序是:
- with 語句 首先執(zhí)行 yield 之前的語句,因此打印出.
- yield 調(diào)用會(huì)執(zhí)行 with 語句內(nèi)部的所有語句,因此打印出 hello 和 world.
- 最后執(zhí)行yield之后的語句,打印出結(jié)果
二、@closing
如果一個(gè)對象沒有實(shí)現(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 也是一個(gè)經(jīng)過 @contextmanager 裝飾的generator
@contextmanager def closing(thing): try: yield thing finally: thing.close()
它的作用就是把任意對象變?yōu)樯舷挛膶ο?,并支?with語句。
到此這篇關(guān)于python內(nèi)置模塊之上下文管理contextlib的文章就介紹到這了。希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
python transpose()處理高維度數(shù)組的軸變換的實(shí)現(xiàn)
本文主要介紹了python transpose()處理高維度數(shù)組的軸變換的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-09-09解決pytorch GPU 計(jì)算過程中出現(xiàn)內(nèi)存耗盡的問題
今天小編就為大家分享一篇解決pytorch GPU 計(jì)算過程中出現(xiàn)內(nèi)存耗盡的問題,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08關(guān)于Python 3中print函數(shù)的換行詳解
最近在學(xué)習(xí)python3,發(fā)現(xiàn)了一個(gè)問題想著總結(jié)出來,所以下面這篇文章主要給大家介紹了關(guān)于Python 3中print函數(shù)換行的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),對需要的朋友們具有一定的參考學(xué)習(xí)價(jià)值,感興趣的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧。2017-08-08Python模塊Typing.overload的使用場景分析
在 Python 中,typing.overload 是一個(gè)用于定義函數(shù)重載的裝飾器,函數(shù)重載是指在一個(gè)類中可以定義多個(gè)相同名字但參數(shù)不同的函數(shù),使得在調(diào)用函數(shù)時(shí)可以根據(jù)參數(shù)的不同選擇不同的函數(shù)執(zhí)行,這篇文章主要介紹了Python模塊Typing.overload的使用,需要的朋友可以參考下2024-02-02python 動(dòng)態(tài)調(diào)用函數(shù)實(shí)例解析
這篇文章主要介紹了python 動(dòng)態(tài)調(diào)用函數(shù)實(shí)例解析,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-10-10python買賣股票的最佳時(shí)機(jī)(基于貪心/蠻力算法)
這篇文章主要介紹了python買賣股票的最佳時(shí)機(jī)(基于貪心/蠻力算法),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-07-07python中數(shù)據(jù)爬蟲requests庫使用方法詳解
本篇文章主要介紹了python中數(shù)據(jù)爬蟲requests庫使用方法詳解,小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過來看看吧2018-02-02