Python でコンテキストマネージャーを使用する方法 (ステートメントあり)

Python でコンテキストマネージャーがどのように機能するかを学びます。 with ステートメントをマスターし、システムリソースを管理し、カスタムコンテキストマネージャーを作成します。

概要

ファイルハンドルを閉じる、データベース接続を切断する、スレッドロックを解放するなど、外部リソースを適切に管理することは、信頼性が高く実稼働対応のソフトウェアを構築するために不可欠です。リソースの解放に失敗すると、メモリリーク、ファイルロックエラー、アプリケーションのハングが発生する可能性があります。

Python の「with」ステートメントは、コンテキストマネージャーを作成することでリソースのクリーンアップを簡素化します。コンテキストマネージャーは、コードが例外をスローしたり早期に終了したりしても、コントロールがブロックを離れるときにクリーンアップアクションが自動的に実行されることを保証します。これにより、冗長で反復的な `try-finally` ブロックが不要になります。

コンテキストマネージャーは一般的に「open()」などの組み込み操作に関連付けられていますが、独自のカスタムコンテキストマネージャーを簡単に設計できます。これらを構築するには、`__enter__` および `__exit__` マジックメソッドを実装するクラスを作成するか、`contextlib.contextmanager` デコレーターで装飾されたジェネレーター関数を作成します。

コードと実行の出力

このスクリプトは、クラス実装と contextlib デコレーターを使用して Python でカスタムコンテキストマネージャーを作成する方法を示します。

context_managers.py

エディターで試してみる

from contextlib import contextmanager

# Method 1: Class-based context manager
class CustomResource:
    def __enter__(self):
        print("[Resource] Opening connection...")
        return self

    def do_work(self):
        print("[Resource] Performing action...")

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("[Resource] Closing connection safely.")
        return False

print("--- Class Context Manager ---")
with CustomResource() as resource:
    resource.do_work()

# Method 2: Generator-based context manager (contextlib)
@contextmanager
def simple_timer():
    print("\n[Timer] Started...")
    try:
        yield
    finally:
        print("[Timer] Finished!")

print("\n--- Decorator Context Manager ---")
with simple_timer():
    print("Performing tasks inside context...")

端子出力

--- Class Context Manager ---
[Resource] Opening connection...
[Resource] Performing action...
[Resource] Closing connection safely.

--- Decorator Context Manager ---

[Timer] Started...
Performing tasks inside context...
[Timer] Finished!