python怎么創(chuàng)建支持with語句的對(duì)象

這篇文章主要為大家展示了“python怎么創(chuàng)建支持with語句的對(duì)象”，內(nèi)容簡而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領(lǐng)大家一起研究并學(xué)習(xí)一下“python怎么創(chuàng)建支持with語句的對(duì)象”這篇文章吧。

創(chuàng)建支持「with」語句的對(duì)象

舉例而言，我們都知道如何使用「with」語句打開文件或獲取鎖，但是我們可以實(shí)現(xiàn)自己上下文表達(dá)式嗎？是的，我們可以使用「enter」和「exit」來實(shí)現(xiàn)上下文管理協(xié)議:

class Connection:  
    def __init__(self):  
        ...  
    def __enter__(self):  
        # Initialize connection...  
    def __exit__(self, type, value, traceback):  
        # Close connection...  
with Connection() as c:  
    # __enter__() executes  
    ...  
    # conn.__exit__() executes

這是在 Python 中最常見的實(shí)現(xiàn)上下文管理的方法，但是還有更簡單的方法:

from contextlib import contextmanager  
@contextmanager  
def tag(name):  
    print(f"<{name}>")  
    yield  
    print(f"</{name}>")  
with tag("h2"):  
    print("This is Title.")

上面這段代碼使用 contextmanager 的 manager 裝飾器實(shí)現(xiàn)了內(nèi)容管理協(xié)議。在進(jìn)入 with 塊時(shí) tag 函數(shù)的第一部分（在 yield 之前的部分）就已經(jīng)執(zhí)行了，然后 with 塊才被執(zhí)行，最后執(zhí)行 tag 函數(shù)的其余部分。

以上是“python怎么創(chuàng)建支持with語句的對(duì)象”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對(duì)大家有所幫助，如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道！