Python上下文管理器實現方法總結

2022-09-27 07:06:08 字數 2278 閱讀 3067

目錄

當你的**邏輯需要用到如下關鍵字時,可以考慮使用上下文管理器讓你的**更加優雅:

try:

...finally:

...接下來介紹實現上下文管理器的三種方法。

總所周知,open()是預設支援上下文管理器的。所以開啟乙個txt檔案,並向裡面寫入內容,再關閉這個檔案的**可以這樣寫:

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中實現with語句的使用,就需要借助上下文管理器協議。也就是需要實現__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 txturckyqliu

writing...

closing the txt

同時能夠看到本地生成了2.txt檔案。需要注意的是,__enter__得return例項物件,不然會報異常:attributeerror: 'nonetype' object has no attribute 'write'

這是因為python中的函式預設返回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裝飾的函式中,需要用yield隔開兩個邏輯語句。這裡yield出來的物件會被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")

selwww.cppcns.comf.f.close()

with closing(openmyfile("4.txt")) as fil程式設計客棧e:

file.write("this is demo4")

# 輸出:

opening the txt

closing the txt

與方法1不同。經過closing()方法包裝過後,在with語句結束時,會強制呼叫物件的close()方法。所以使用方法3時,需要定義的方法不是__exit__()而是close()。

python 上下文管理器實現

上下文管理器實現需要有 enter 和 exit 的特殊方法 enter self 進入上下文管理器時呼叫此方法,其返回值將被放入with as語句中as說明符指定的變數中。exit self,type,value,tb 離開上下文管理器呼叫此方法。如果有異常出現,type value tb分別為異...

python 上下文管理器

上下文管理器允許你在有需要的時候,精確地分配和釋放資源。使用上下文管理器最廣泛的案例就是with語句了。想象下你有兩個需要結對執行的相關操作,然後還要在它們中間放置一段 上下文管理器就是專門讓你做這種事情的。舉個例子 with open some file w as opened file open...

python上下文管理器

上下文管理器是乙個包裝任意 塊的物件。上下文管理器保證進入上下文管理器時,每次 執行的一致性 當退出上下文管理器時,相關資源會被正確 這裡被正確 指的是在 exit 方法自定義 比如關閉資料庫游標 值得注意的是,上下文管理器一定能夠保證退出步驟的執行。如果進入上下文管理器,根據定義,一定會有退出步驟...