Pythonのコンテキストマネージャーとは？__enter__と__exit__の使い方を解説

Pythonでは、with文によってランタイムコンテキスト（実行時コンテキスト）がサポートされています。このコンテキストは「コンテキストマネージャー」によって定義され、ユーザーが独自にクラスを作成して、実行時のコンテキストを自由に定義することも可能です。

コンテキストマネージャーは、処理本体（ステートメントボディ）を実行する前にコンテキストへ入り、処理が完了した時点で終了します。これにより、リソースの確保と解放を安全かつ確実に行うことができます。

コンテキストマネージャーを実装するためには、次の2つの特殊メソッドを使用します。

__enter__() メソッド

__enter__() メソッドは、ランタイムコンテキストに「入る」際に呼び出されるメソッドです。戻り値として、現在のオブジェクト自身、または関連する別のオブジェクトを返します。この戻り値は、with文の as 句で指定した識別子にバインドされます。

__exit__(exc_type, exc_val, exc_tb) メソッド

__exit__() メソッドは、ランタイムコンテキストを「抜ける」際に呼び出され、Boolean型（真偽値）の結果を返します。この戻り値は、例外が発生したかどうかを示すものです。with文の処理中に例外が発生した場合、その情報は exc_type（例外の型）、exc_val（例外の値）、exc_tb（トレースバック）として渡され、適切に後処理を行うことができます。

サンプルコード

以下は、ファイル操作を安全に行うカスタムコンテキストマネージャーの例です。ファイルを開く処理を __enter__() で、クローズ処理を __exit__() で行っています。

class MyFileManager:
    def __init__(self, fname):
        self.file_name = fname

    def __enter__(self):
        self.myFile = open(self.file_name, 'r')
        return self.myFile

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.myFile:
            self.myFile.close()

with MyFileManager('sampleTextFile.txt') as myFile:
    x = myFile.read()
print(x)

実行結果

Test File.
We can store different contents in this file
~!@#$%^&*()_+/*-+\][{}|:;"'<.>/,'"]

このように、コンテキストマネージャーを使うことで、ファイルのクローズ忘れなどのリソースリークを防ぎ、例外が発生した場合でも確実に後処理を実行できる、堅牢なコードを書くことができます。