1import time2 3class Timer:4 def __enter__(self):5 self.start = time.time()6 return self7 8 def __exit__(self, *args):9 self.end = time.time()10 self.elapsed = self.end - self.start11