當我們在編寫代碼時，經常會遇到需要管理資源的情況，比如打開和關閉文件，如果遇到了一些異常情況，我們需要關閉資源，不然會導致資源泄露，雖然我們可以通過手動的方式來關閉，但如果有多個異常情況需要考慮的話，萬一不小心漏了一處，就芭比Q了。所以，如果有一種更加優雅的方式來處理資源泄露的問題，那必定是非常ni ...
當我們在編寫代碼時，經常會遇到需要管理資源的情況，比如打開和關閉文件，如果遇到了一些異常情況，我們需要關閉資源，不然會導致資源泄露，雖然我們可以通過手動的方式來關閉，但如果有多個異常情況需要考慮的話，萬一不小心漏了一處，就芭比Q了。所以，如果有一種更加優雅的方式來處理資源泄露的問題，那必定是非常nice的。而上下文管理器就是在這樣的背景下誕生的。
上下文管理器
上下文管理器（Context Manager）是 Python 中用來管理資源、執行特定操作以及處理異常的對象。上下文管理器通過實現特定的方法使得它們可以在 with 語句中被使用，確保資源的正確獲取和釋放。
在 Python 中，上下文管理器通常使用類中的特殊方法 __enter__ 和 __exit__ 來實現。具體來說：
1、 __enter__ 方法在進入 with 語句塊時被調用，通常用於獲取資源或者執行一些初始化操作，一般返回管理器對象。
2、 __exit__ 方法在離開 with 語句塊時被調用，無論代碼塊中是否發生異常，該方法都會被執行。這個方法通常用來釋放資源，清理工作以及處理異常情況。放法中含有三個參數分別是：異常類型、異常實例對象和異常棧對象
class MyContextManager:
def __enter__(self):
print("Entering the context")
# 執行一些初始化操作，比如打開資源
return self
def __exit__(self, exc_type, exc_value, traceback):
print("Exiting the context")
# 執行一些清理操作，比如關閉資源
if exc_type is not None:
print(f"Exception: {exc_type}, {exc_value}")
# 返回 True 表示異常已經被處理，返回 False 則異常會被傳播
return True
# 使用上下文管理器
with MyContextManager() as manager:
print("Inside the context")
# 執行一些操作，比如讀寫資源
# 如果在這裡發生了異常，異常會被傳遞給 __exit__ 方法進行處理
比如上面的這個示例中，MyContextManager 是一個簡單的上下文管理器類，通過實現 __enter__ 和 __exit__ 方法來定義上下文管理器的行為。在 with 語句塊中使用該上下文管理器，會執行相應的操作（進入、執行、退出上下文），確保資源的正確管理。
上面的代碼運行結果：
Entering the context
Inside the context
Exiting the context
with語句
很多人說with語句是用來創建上下文管理器，準確的說，這樣子的說法是不對的。之前的例子中，MyContextManager是一個上下文管理器的類，而我們所謂的創建上下文管理器，無非就是創建上下文管理器的對象。那麼以下這樣子不就是創建了一個上下文管理器對象嗎？
m = MyContextManager()
所以準確的說，with語句是Python提供的一種管理上下文的語法結構，能夠自動調用上下文管理器的__enter__和__exit__方法。通過with語句，可以確保在進入和退出代碼塊時自動執行必要的操作。
例如，使用with語句的上下文管理器來打開和關閉文件：
with open('example.txt', 'r') as file:
content = file.read()
print(content)
# 文件在with語句塊結束後會自動關閉，即使發生異常也會正確關閉文件
open函數返回的文件對象是一個上下文管理器，本身就不需要再去實現了，通過使用with關鍵字，確保在代碼塊執行完畢後文件會被正確關閉。無需手動調用 file.close() 方法。這樣可以提高代碼可讀性和可維護性，避免忘記關閉文件而導致資源泄漏的問題。
當然我們也可以自己手動實現一個file上下文管理器
class FileContextManager:
def __init__(self, file_name, mode):
self.file_name = file_name
self.mode = mode
def __enter__(self):
self.file = open(self.file_name, self.mode)
return self.file
def __exit__(self, exc_type, exc_value, exc_traceback):
if self.file:
self.file.close()
# 使用自定義的文件上下文管理器類來打開文件
with FileContextManager('example.txt', 'w') as file:
file.write("程式員一點來啦！")
這個就是一個比較簡單的文件上下文管理器，FileContextManager('example.txt', 'w')和上面open('example.txt', 'r')功能是差不多的。
以上就是關於上下文管理器的一些簡單介紹。總的來說，上下文管理器是 Python 提供的一種非常方便的資源管理機制，可以幫助我們更好地管理和釋放資源，簡化代碼邏輯，提高代碼的健壯性，寫出更加優雅的代碼，在項目開發中使用起來非常的舒服。
更多精彩內容，請關註同名公眾：一點sir（alittle-sir）
