python如何使用ContextLib

小編給大家分享一下python如何使用ContextLib，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

上下文管理庫(ContextLib)

contextlib模塊包含了與上下文管理器和with聲明相關的工具。通常如果你想寫一個上下文管理器，則你需要定義一個類包含enter方法以及exit方法，例如：

import timeclass demo:

  def __init__(self, label):    self.label = label  def __enter__(self):    self.start = time.time()  def __exit__(self, exc_ty, exc_val, exc_tb):    end = time.time()

    print('{}: {}'.format(self.label, end - self.start))

完整的例子在此：

import timeclass demo:

  def __init__(self, label):    self.label = label  def __enter__(self):    self.start = time.time()  def __exit__(self, exc_ty, exc_val, exc_tb):    end = time.time()

    print('{}: {}'.format(self.label, end - self.start))

with demo('counting'):

  n = 10000000

  while n > 0:

    n -= 1# counting: 1.36000013351

上下文管理器被with聲明所激活，這個API涉及到兩個方法。 enter方法，當執行流進入with代碼塊時，enter方法將執行。并且它將返回一個可供上下文使用的對象。
當執行流離開with代碼塊時，exit方法被調用，它將清理被使用的資源。

利用@contextmanager裝飾器改寫上面那個例子：

from contextlib import contextmanagerimport time@contextmanagerdef demo(label):

  start = time.time()  try:    yield

  finally:

    end = time.time()

    print('{}: {}'.format(label, end - start))with demo('counting'):

  n = 10000000

  while n > 0:

    n -= 1# counting: 1.32399988174

看上面這個例子，函數中yield之前的所有代碼都類似于上下文管理器中enter方法的內容。而yield之后的所有代碼都如exit方法的內容。如果執行過程中發生了異常，則會在yield語句觸發。

以上是“python如何使用ContextLib”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！