python內置模塊之上下文管理contextlib
Python中當我們們打開文本時,通常會是用with語句,with語句允許我們非常方便的使用資源,而不必擔心資源沒有關閉。
with open('/path/filename', 'r') as f: f.read()
然而,并不是只有open()函數返回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()
一、@contextmanager
編寫 __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("h1"): print("hello") print("world")
上述代碼執(zhí)行結果:
hello
world
</h1>
代碼的執(zhí)行順序是:
- with 語句 首先執(zhí)行 yield 之前的語句,因此打印出.
- yield 調用會執(zhí)行 with 語句內部的所有語句,因此打印出 hello 和 world.
- 最后執(zhí)行yield之后的語句,打印出結果
二、@closing
如果一個對象沒有實現(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 也是一個經過 @contextmanager 裝飾的generator
@contextmanager def closing(thing): try: yield thing finally: thing.close()
它的作用就是把任意對象變?yōu)樯舷挛膶ο?,并支?with語句。
到此這篇關于python內置模塊之上下文管理contextlib的文章就介紹到這了。希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關文章
python transpose()處理高維度數組的軸變換的實現(xiàn)
本文主要介紹了python transpose()處理高維度數組的軸變換的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2023-09-09解決pytorch GPU 計算過程中出現(xiàn)內存耗盡的問題
今天小編就為大家分享一篇解決pytorch GPU 計算過程中出現(xiàn)內存耗盡的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08Python模塊Typing.overload的使用場景分析
在 Python 中,typing.overload 是一個用于定義函數重載的裝飾器,函數重載是指在一個類中可以定義多個相同名字但參數不同的函數,使得在調用函數時可以根據參數的不同選擇不同的函數執(zhí)行,這篇文章主要介紹了Python模塊Typing.overload的使用,需要的朋友可以參考下2024-02-02