前言

with語句的使用給我們帶來了很多的便利，最常用的可能就是關閉一個文件，釋放一把鎖。

既然with語句這麼好用，那我也想讓我自己寫的代碼也能夠使用with語句，該怎麼實現？

下麵具體介紹怎樣實現一個自己的with語句

使用類實現

要想使用with語句，那就要遵循with語句的使用規矩，也就是上下文管理器協議

這個協議提示我們要在類中去實現兩個特殊方法,enter(self)和exit(self, exc_type, exc_val, exc_tb)

有了這兩個方法，就可以使用with語句執行了，它的執行過程為先執行enter(self)方法，然後執行你添加在with下的代碼，然後最後執行exit方法

如果在執行的過程中出現了異常，那麼會自動執行exit方法，

class ErrorProduce:
    def __init__(self):
        pass

    def produce(self):
        raise RuntimeError('ERROR')

class MyWith:
    def __enter__(self):
        return ErrorProduce()

    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type:
            print("有錯")
            #print(exc_val)
            #return True
        else:
            print("完美執行")

with MyWith() as E:
    E.produce()

將上述代碼運行，會先列印出“有錯”，然後再列印出相關錯誤，然後程式由於錯誤終止，也就是說，在產生了錯誤後，還是執行了exit中的代碼

如果將exit中的兩個註釋符號去掉的話，那麼再次運行程式不會報錯，但會將exc_val中的錯誤信息列印出來，這裡相當於通過exit方法將錯誤抹除掉了，抹除方式為返回True，但也把with中拋出錯誤後的所有代碼都跳過了。

使用函數實現

通過引入contextlib模塊中的contextmanager裝飾器，裝飾函數使函數可以使用with語句

from contextlib import contextmanager

class ErrorProduce:
    def __init__(self):
        pass

    def produce(self):
        raise RuntimeError('ERROR')

@contextmanager
def my_with():
    try:
        yield ErrorProduce()
    except Exception as e:
        print(e)
　　　　 #raise
    finally:
        print('收尾')

with my_with() as E:
    E.produce()
    print('aa')

函數中使用了yield，那麼它也就變成了生成器，通過yield來劃分相當於類中的enter和exit，yield返回的值也就是可以用作as後的值

上段代碼中我們發現使用了try，except，finally異常處理的語句，這些語句是必要的嗎？

通過測試，如果沒有這些異常處理，那麼with語句還是可以執行的，這裡的意思是可以使用with語句，不會報不能使用with的錯誤，但是使用後，只有with下的語句中沒有拋出異常，那麼yield語句後邊的代碼才會執行，一旦有異常拋出，仍然是程式運行終止。所以，一般還是要配合try，except，finally語句來構造。

還有一點要註意，使用這種方式，異常被except捕捉後，那麼這個異常就失效了，一般的我們還是會需要在except中再次將捕捉的異常拋出，也就是在後面加上raise，因為我們使用with只是為了做好一個收尾，如關閉文件句柄。或者另一種方式，我們乾脆就不寫except語句，直接寫finally語句了。

• 文章中出現的棕色加粗的單詞，是由於markdown-here將特殊方法兩邊的下劃線轉換產生的，原意為特殊方法