首頁  >  文章  >  後端開發  >  揭秘 Python 中的 with 關鍵字

揭秘 Python 中的 with 關鍵字

PHPz
PHPz轉載
2023-04-14 11:31:061134瀏覽

揭秘 Python 中的 with 關鍵字

我們中的許多人在Python 程式碼中一遍又一遍地看到這個程式碼片段:

with open('Hi.text', 'w') as f:
f.write("Hello, there")

但是,我們中的一些人不知道with 有什麼用,以及為什麼我們需要在這裡使用它。在此閱讀中,您將找到關於 with 可解決的幾乎所有問題。讓我們開始吧!

首先,讓我們考慮一下如果不使用 with 關鍵字我們需要做什麼。在這種情況下,我們需要先開啟檔案並嘗試執行 write。不管成功與否,我們最好在最後關閉它,所以我們的程式碼將如下所示:

f = open('Hi.text', 'w')
try:
f.write('Hello, there')
finally:
f.close()

那麼,with 關鍵字有什麼用呢?它只是有助於將我們的 try..finally 程式碼縮短為 with... 的單一語句!這就是 with 語句用法。

那麼,它到底是什麼?事實上,with 語句本身在 Python 中並沒有什麼特別之處,它只是 Python 中上下文管理器的一個特性。上下文管理器,引用自Python 官方文檔,是一種讓您在需要時準確分配和釋放資源的方法,或者簡單來說:當您在某些資源上做某事時縮短您的程式碼片段,這意味著您可以自己定義with 語句的用法!

我們如何做到這一點?嗯,很簡單,你只需要實作兩個魔術函數:一個叫做 __enter__,另一個叫做 __exit__。第一種方法是寫一個實作這兩個函數的類,如下所示:

class My_file:
def __init__(self, fname):
self.fname = fname
def __enter__(self):
self.file = open(self.fname, 'w')
return self.file
def __exit__(self, exc_type, exc_val, exc_trace_back):
if self.file:
self.file.close()

在這裡,我們創建了一個普通的 Python 類,實作了兩個魔術函數。注意這兩個函數的簽章:__enter__ 只接受 self ,而 __exit__ 將接受更多參數,範例中的這三個是標準形式。這樣,我們就可以直接使用:

with My_file('hello.txt') as f:
f.write('hello, world!')

這裡的 with 語句會先呼叫 __init__ 建構一個新對象,然後再呼叫 __enter__ 方法;最後,它會在程式碼區塊完成之前觸發 __exit__ 方法。所以,上面程式碼的大致等價如下:

myfile = My_file('hello.txt')
f = myfile.__enter__()
f.write('hello, world!')
myfile.__exit(...)

實作上下文管理器的第二種方法是透過裝飾器,如下:

  1. 你import contextmanager from contextlib
  2. ## 你寫一個函數來實作你想要的with語句。
在函數上方加上一個裝飾器@contextmanager。

使用你的with your_function!

根據上面的介紹,讓我們寫一個裝飾器上下文管理器!

from contextlib import contextmanager
@contextmanager
def my_file_open(fname):
try:
f = open(fname, 'w')
yield f
finally:
print('Closing file')
f.close()
with file_open('hi.txt') as f:
f.write('hello world')

以上幾乎涵蓋了 with 語句和上下文管理器的所有基礎知識,但是如果您想了解更多信息,請繼續...!

contextlib 中有一些方便的工具供您使用。第一個是 closing 。 closing 基本上會用你在退出之前實現的另一個函數來包裝你的程式碼區塊。

@contextmanager
def closing(f):
try:
f.write("Finish writing")
finally:
f.close()
with closing(open("hi.text")):
f.write("hello world")

例如,在上面的程式碼中,我們可以直接呼叫close(your_way_of_getting_resource) ,在你下面寫的程式碼區塊即將完成之前(f.write("hello world")),它會執行try ..finally 我們在上面定義的區塊。

另一個是使用 suppress 工具。我們知道,在很多情況下,如果我們嘗試取得一些資源,很可能在開啟檔案時會出現 FileNotFoundException 等錯誤。在某些情況下,我們希望捕獲錯誤或抑制錯誤,以便程式繼續正常運作。 suppress 是我們可以抑制警告的一種方式。你需要做的就是弄清楚你想要抑制哪個異常,並寫with suppress(your_choice_of_exception),Python 將從這裡開始處理它。

在其他情況下,您可能只想在輸入 with 程式碼區塊時執行某些操作。在這種情況下,nullcontext 對你來說會很方便。 nullcontext 只會傳回你在 __enter__ 函數中定義的東西,而不會做任何其他事情。如果您在 Python 中處理 async 操作以存取資源,則 aclosure 是處理這種情況的實用工具。

###總結######本文介紹了 with 語句的一些基本概念和用法及其底層工作原理。還有很多有趣的東西,請查看 Python 的 contextlib 文件。最後,祝您能像往常一樣快樂學習和快樂編碼! ###

以上是揭秘 Python 中的 with 關鍵字的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:51cto.com。如有侵權,請聯絡admin@php.cn刪除