pythonでwith context内での実行の実行時間を計測するスクリプト

elapse_time_in_with_context.py

# with context内での実行の実行時間を計測するスクリプト
import time

class ElapseTimeInWithContext(object):
    def __init__(self, context):
      self.context = context
    def __enter__(self):
      self.start = time.time()
      return self
    def __exit__(self, *args):
      print(f"{self.context} elapsed time (msec): {1000 * (time.time() - self.start)}")