Python上下文管理器是一種特殊的對象,它允許你在執行代碼塊之前和之后執行一些操作
上下文管理器的工作原理是通過實現兩個特殊的方法:__enter__()
和 __exit__()
。當你使用 with
語句時,這兩個方法會按照以下順序被調用:
__enter__()
:在執行 with
語句塊中的代碼之前調用。這個方法應該返回一個對象,這個對象將賦值給 as
子句中的變量(如果有的話)。
with
語句塊中的代碼:在這里執行你需要的操作。
__exit__(exc_type, exc_value, traceback)
:在執行完 with
語句塊中的代碼之后調用。這個方法用于處理在 with
語句塊中發生的異常。如果 __exit__()
方法返回 True
,則異常會被忽略;如果返回 False
或者沒有返回值,則異常會被重新拋出。
下面是一個簡單的上下文管理器的例子,用于計算代碼塊的執行時間:
import time
class Timer:
def __enter__(self):
self.start = time.time()
return self
def __exit__(self, exc_type, exc_value, traceback):
end = time.time()
print(f"Elapsed time: {end - self.start} seconds")
with Timer() as t:
# 在這里執行你需要的操作
time.sleep(2)
在這個例子中,Timer
類實現了上下文管理器。當你使用 with Timer() as t:
語句時,__enter__()
方法會被調用,記錄開始時間,并將 self
賦值給變量 t
。然后執行 with
語句塊中的代碼(在這里是 time.sleep(2)
)。最后,__exit__()
方法會被調用,計算并打印出經過的時間。