深人了解Python上下文管理器
下面先來介紹一下with關(guān)鍵字在文件讀寫中的應(yīng)用，簡(jiǎn)單了解上下文管理器的功能。
with語句
在Python文件及目錄處理方法中介紹了讀寫大文件建議使用with語句，with語句會(huì)進(jìn)行資源的自動(dòng)管理。文件很多的情況下也會(huì)導(dǎo)致資源泄露，下面來打開100000個(gè)文件，不進(jìn)行文件關(guān)閉操作：
for x in range(100000):
file = open('test.txt', 'w')
file_descriptors.append(file)
執(zhí)行會(huì)報(bào)如下錯(cuò)誤：
OSError: [Errno 24] Too many open files: 'test.txt'
原因就是打開了太多文件而沒有及時(shí)關(guān)閉導(dǎo)致了資源泄露，造成系統(tǒng)崩潰。完成處理后需要對(duì)文件進(jìn)行關(guān)閉操作：
file_descriptors = []
for x in range(10000):
file = open('test.txt', 'w')
try:
file_descriptors.append(file)
finally:
file.close()
使用 with 語句可以完成自動(dòng)分配并且釋放資源，比上面的寫法更加簡(jiǎn)潔：
file_descriptors = []
for x in range(10000):
with open('test.txt', 'w') as file:
file_descriptors.append(file)
上下文管理器創(chuàng)建
基于類的上下文管理器
可以使用類來創(chuàng)建上下文管理器，需要保證這個(gè)類包括兩個(gè)方法：__enter__() 和__exit__()。其中，方法 __enter__() 返回需要被管理的資源，方法 __exit__() 進(jìn)行資源釋放、清理操作。
下面來模擬 Python 的打開、關(guān)閉文件操作：
class FileManager:
def __init__(self, name, mode):
print('__init__ method called')
self.name = name
self.mode = mode
self.file = None
def __enter__(self):
print('__enter__ method called')
self.file = open(self.name, self.mode)
return self.file
def __exit__(self, exc_type, exc_value, exc_traceback):
print('__exit__ method called')
if self.file:
self.file.close()
if exc_type:
print(f'exc_type: {exc_type}')
print(f'exc_value: {exc_value}')
print(f'exc_traceback: {exc_traceback}')
return True
with FileManager('test.txt', 'w') as f:
print('開始寫操作')
f.write('hello world !')
print(f.closed)
執(zhí)行結(jié)果：
__init__ method called
__enter__ method called
開始寫操作
__exit__ method called
exc_type: <class 'Exception'>
exc_value: exception raised
exc_traceback: <traceback object at 0x000001B43C2444C8>
True
可以看到執(zhí)行順序?yàn)椋?/strong>
__init__()：初始化對(duì)象 FileManager__enter__()：打開文件，返回 FileManager 對(duì)象
with中的代碼
__exit__()：關(guān)閉打開的文件流
__exit__()方法中的參數(shù)exc_type, exc_value, 和 exc_traceback 用于管理異常。
@contextmanager 裝飾器
可以使用 contextlib.contextmanager 裝飾器而不使用類的方式來實(shí)現(xiàn)上下文管理器，它是基于生成器的上下文管理器，用以支持 with 語句。
仍以打開、關(guān)閉文件為例：
from contextlib import contextmanager
@contextmanager
def file_manager(name, mode):
try:
f = open(name, mode)
yield f
finally:
f.close()
with file_manager('test.txt', 'w') as f:
f.write('hello world !')
其中 file_manager() 函數(shù)是一個(gè)生成器，yield 之前可以看成是__enter__方法中的內(nèi)容，yield 后面的是 __exit__() 內(nèi)容。加上@contextmanager裝飾器，使用基于生成器的上下文管理器時(shí)，不需要定義__enter__()和__exit__()方法。
總結(jié)
上下文管理器可確保用過的資源得到迅速釋放，通常和 with 語句一起使用，大大提高了程序的簡(jiǎn)潔度。另外需要注意的是，編寫基于類或者生成器的上下文管理器時(shí)，記住不要忘記釋放資源。--THE END--
本篇文章就到這里了，希望能夠給你帶來幫助，也希望您能夠多多關(guān)注腳本之家的更多內(nèi)容!
相關(guān)文章
這篇文章主要介紹了如何使用Python?繪制瀑布圖，我們一起了解瀑布圖的重要性，以及如何使用不同的繪圖庫繪制瀑布圖。瀑布圖是一種二維圖表，專門用于了解隨著時(shí)間或多個(gè)步驟或變量的增量正負(fù)變化的影響，下文更多詳細(xì)內(nèi)容需要的小伙伴可以參考一下2022-05-05
Python數(shù)據(jù)結(jié)構(gòu)與算法之算法分析詳解
算法分析的主要目標(biāo)是從運(yùn)行時(shí)間和內(nèi)存空間消耗等方面比較算法。本文將為大家詳細(xì)介紹Python數(shù)據(jù)結(jié)構(gòu)與算法中的算法分析，需要的可以參考一下2021-12-12
這篇文章主要介紹了Pytorch中的數(shù)據(jù)集劃分&正則化方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2021-05-05
postman模擬訪問具有Session的post請(qǐng)求方法
今天小編就為大家分享一篇postman模擬訪問具有Session的post請(qǐng)求方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-07-07
在Python中將函數(shù)作為另一個(gè)函數(shù)的參數(shù)傳入并調(diào)用的方法
今天小編就為大家分享一篇在Python中將函數(shù)作為另一個(gè)函數(shù)的參數(shù)傳入并調(diào)用的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-01-01
這篇文章主要介紹了python協(xié)程與?asyncio?庫詳情，文章基于python展開詳細(xì)內(nèi)容，具有一定的參考價(jià)值。需要的小伙伴可以參考一下2022-05-05
python自然語言編碼轉(zhuǎn)換模塊codecs介紹
這篇文章主要介紹了python自然語言編碼轉(zhuǎn)換模塊codecs介紹,codecs專門用作編碼轉(zhuǎn)換,通過它的接口是可以擴(kuò)展到其他關(guān)于代碼方面的轉(zhuǎn)換,需要的朋友可以參考下2015-04-04
這篇文章主要介紹了Python查詢IP地址歸屬完整代碼，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2017-06-06
解決安裝Python的第三方庫pandas報(bào)錯(cuò)問題
這篇文章主要介紹了解決安裝Python的第三方庫pandas報(bào)錯(cuò)問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-08-08
這篇文章主要為大家詳細(xì)介紹了Python實(shí)現(xiàn)漸變色的水平堆疊圖，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-04-04