sdk/python/arvados/timer.py

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: Apache-2.0
   4
   5 import time
   6
   7 class Timer(object):
   8     def __init__(self, verbose=False):
   9         self.verbose = verbose
  10
  11     def __enter__(self):
  12         self.start = time.time()
  13         return self
  14
  15     def __exit__(self, *args):
  16         self.end = time.time()
  17         self.secs = self.end - self.start
  18         self.msecs = self.secs * 1000  # millisecs
  19         if self.verbose:
  20             print('elapsed time: %f ms' % self.msecs)