當(dāng)你的代碼邏輯需要用到如下關(guān)鍵字時(shí)，可以考慮使用上下文管理器讓你的代碼更加優(yōu)雅：
try: ... finally: ...
接下來(lái)介紹實(shí)現(xiàn)上下文管理器的三種方法。
總所周知，open()是默認(rèn)支持上下文管理器的。所以打開(kāi)一個(gè)txt文件，并向里面寫入內(nèi)容，再關(guān)閉這個(gè)文件的代碼可以這樣寫：
with open("1.txt", "w") as file:
file.write("this is a demo")
這是等同于：
file = None
try:
file = open("1.txt", "w")
file.write("this is a demo")
finally:
file.close()
要在Python中實(shí)現(xiàn)with語(yǔ)句的使用，就需要借助上下文管理器協(xié)議。也就是需要實(shí)現(xiàn)__enter__和__exit__兩個(gè)魔法方法。
class OpenMyFile(object):
def __init__(self, path):
self.path = path
def __enter__(self):
print("opening the txt")
self.f = open(self.path, "w")
return self
def __exit__(self, *args, **kwargs):
print("closing the txt")
self.f.close()
def write(self, string):
print("writing...")
self.f.write(string)
with OpenMyFile("2.txt") as file:
file.write("this is a demo2")
# 輸出：
opening the txt
writing...
closing the txt
同時(shí)能夠看到本地生成了2.txt文件。需要注意的是，__enter__得return實(shí)例對(duì)象，不然會(huì)報(bào)異常：AttributeError: 'NoneType' object has no attribute 'write'
這是因?yàn)镻ython中的函數(shù)默認(rèn)返回None。
利用contextlib中的contextmanager裝飾器。
from contextlib import contextmanager
@contextmanager
def open_my_file(path):
print("opening the txt")
f = open("3.txt", "w")
yield f
print("closing the txt")
f.close()
with open_my_file("3.txt") as file:
file.write("this is demo3")
# 輸出：
opening the txt
closing the txt
在@contextmanager裝飾的函數(shù)中，需要用yield隔開(kāi)兩個(gè)邏輯語(yǔ)句。這里yield出來(lái)的對(duì)象會(huì)被as后面的變量接收。
利用contextlib中的closing()方法。
from contextlib import closing
class OpenMyFile(object):
def __init__(self, path):
print("opening the txt")
self.f = open(path, "w")
def write(self, string):
self.f.write(string)
def close(self):
print("closing the txt")
self.f.close()
with closing(OpenMyFile("4.txt")) as file:
file.write("this is demo4")
# 輸出：
opening the txt
closing the txt
與方法1不同。經(jīng)過(guò)closing()方法包裝過(guò)后，在with語(yǔ)句結(jié)束時(shí)，會(huì)強(qiáng)制調(diào)用對(duì)象的close()方法。所以使用方法3時(shí)，需要定義的方法不是__exit__()而是close()。
到此這篇關(guān)于Python上下文管理器實(shí)現(xiàn)方法總結(jié)的文章就介紹到這了,更多相關(guān)Python上下文管理器實(shí)現(xiàn)的三種方法內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
標(biāo)簽：盤錦 佳木斯 宜昌 珠海 上饒 湖北 西寧 潮州
巨人網(wǎng)絡(luò)通訊聲明：本文標(biāo)題《Python上下文管理器實(shí)現(xiàn)方法總結(jié)》，本文關(guān)鍵詞 Python,上下文,管理器,實(shí)現(xiàn),；如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問(wèn)題，煩請(qǐng)?zhí)峁┫嚓P(guān)信息告之我們，我們將及時(shí)溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò)，涉及言論、版權(quán)與本站無(wú)關(guān)。
