目錄什么時候可以考慮上下文管理器方法1（上下文管理器協(xié)議）方法2（@contextmanager）方法3（contextlib.closing()）什么時候可以考慮上下文管理器

當(dāng)你的代碼邏輯需要用到如下關(guān)鍵字時，可以考慮使用上下文管理器讓你的代碼更加優(yōu)雅：

try:...finally:...

接下來介紹實現(xiàn)上下文管理器的三種方法。

方法1（上下文管理器協(xié)議）

總所周知，open()是默認(rèn)支持上下文管理器的。所以打開一個txt文件，并向里面寫入內(nèi)容，再關(guān)閉這個文件的代碼可以這樣寫：

with open('1.txt', 'w') as file:file.write('this is a demo')

這是等同于：

file = Nonetry: file = open('1.txt', 'w') file.write('this is a demo')finally: file.close()

要在Python中實現(xiàn)with語句的使用，就需要借助上下文管理器協(xié)議。也就是需要實現(xiàn)__enter__和__exit__兩個魔法方法。

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 txtwriting...closing the txt

同時能夠看到本地生成了2.txt文件。需要注意的是，__enter__得return實例對象，不然會報異常：AttributeError: ’NoneType’ object has no attribute ’write’

這是因為Python中的函數(shù)默認(rèn)返回None。

方法2（@contextmanager）

利用contextlib中的contextmanager裝飾器。

from contextlib import contextmanager@contextmanagerdef 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 txtclosing the txt

在@contextmanager裝飾的函數(shù)中，需要用yield隔開兩個邏輯語句。這里yield出來的對象會被as后面的變量接收。

方法3（contextlib.closing()）

利用contextlib中的closing()方法。

from contextlib import closingclass 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 txtclosing the txt

與方法1不同。經(jīng)過closing()方法包裝過后，在with語句結(jié)束時，會強制調(diào)用對象的close()方法。所以使用方法3時，需要定義的方法不是__exit__()而是close()。

到此這篇關(guān)于Python上下文管理器實現(xiàn)方法總結(jié)的文章就介紹到這了,更多相關(guān)Python上下文管理器實現(xiàn)的三種方法內(nèi)容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)！